#include "bits/stdc++.h"
using namespace std;
const int MAXN = 100'001, MS = 18;
const long long INF = 1e18;
int n, s, q, e;
vector<pair<int, long long>> adj[MAXN];
vector<pair<int, int>> edge;
bool shop[MAXN];
int h[MAXN];
long long dist[MAXN];
long long mn[MS][MAXN];
int jmp[MS][MAXN];
void dfs(int u, int p, int ht) {
if (shop[u]) mn[0][u] = 0;
for (auto v : adj[u]) {
if (v.first == p) continue;
dist[v.first] = dist[u] + v.second;
h[v.first] = ht + 1;
jmp[0][v.first] = u;
dfs(v.first, u, ht + 1);
mn[0][u] = min(mn[0][u], mn[0][v.first] + v.second);
}
}
int main() {
ios::sync_with_stdio(0);
cin.tie(0);
cin >> n >> s >> q >> e;
for (int i = 0; i < n - 1; i++) {
int a, b;
long long w;
cin >> a >> b >> w;
edge.push_back({a, b});
adj[a].push_back({b, w});
adj[b].push_back({a, w});
}
for (int i = 0; i < s; i++) {
int a;
cin >> a;
shop[a] = true;
}
for (int i = 1; i <= n; i++) mn[0][i] = INF;
mn[0][e] = 0;
dfs(e, 0, 0);
for (int i = 1; i <= n; i++) {
if (mn[0][i] != INF)
mn[0][i] -= dist[i];
}
for (int i = 1; i < MS; i++) {
for (int j = 1; j <= n; j++) {
jmp[i][j] = jmp[i - 1][jmp[i - 1][j]];
mn[i][j] = min(mn[i - 1][j], mn[i - 1][jmp[i - 1][j]]);
}
}
auto sameSubtree = [&](int a, int b) {
int diff = h[a] - h[b];
for (int i = MS - 1; i >= 0; i--)
if (diff & (1 << i))
a = jmp[i][a];
return a == b;
};
while (q--) {
int a, v;
cin >> a >> v;
a--;
int u = edge[a].first;
if (h[u] < h[edge[a].second]) u = edge[a].second;
if (h[u] <= h[v] && sameSubtree(v, u)) {
long long initDist = dist[v];
long long ans = INF;
int diff = h[v] - h[u];
for (int i = MS - 1; i >= 0; i--) {
if (diff & (1 << i)) {
ans = min(ans, mn[i][v]);
v = jmp[i][v];
}
}
ans = min(ans, mn[0][u]);
if (ans == INF) {
cout << "oo\n";
} else {
cout << initDist + ans << '\n';
}
} else {
cout << "escaped\n";
}
}
}
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 2 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 3 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 4 |
Correct |
3 ms |
2972 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 2 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 3 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 4 |
Correct |
3 ms |
2972 KB |
Output is correct |
| 5 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 6 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 7 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 8 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 9 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 10 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 11 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 12 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 13 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 14 |
Correct |
2 ms |
3156 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
83 ms |
35468 KB |
Output is correct |
| 2 |
Correct |
91 ms |
35236 KB |
Output is correct |
| 3 |
Correct |
107 ms |
35480 KB |
Output is correct |
| 4 |
Correct |
100 ms |
37824 KB |
Output is correct |
| 5 |
Correct |
93 ms |
37888 KB |
Output is correct |
| 6 |
Correct |
141 ms |
40384 KB |
Output is correct |
| # |
결과 |
실행 시간 |
메모리 |
Grader output |
| 1 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 2 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 3 |
Correct |
3 ms |
3028 KB |
Output is correct |
| 4 |
Correct |
3 ms |
2972 KB |
Output is correct |
| 5 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 6 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 7 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 8 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 9 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 10 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 11 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 12 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 13 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 14 |
Correct |
2 ms |
3156 KB |
Output is correct |
| 15 |
Correct |
83 ms |
35468 KB |
Output is correct |
| 16 |
Correct |
91 ms |
35236 KB |
Output is correct |
| 17 |
Correct |
107 ms |
35480 KB |
Output is correct |
| 18 |
Correct |
100 ms |
37824 KB |
Output is correct |
| 19 |
Correct |
93 ms |
37888 KB |
Output is correct |
| 20 |
Correct |
141 ms |
40384 KB |
Output is correct |
| 21 |
Correct |
86 ms |
34876 KB |
Output is correct |
| 22 |
Correct |
79 ms |
34644 KB |
Output is correct |
| 23 |
Correct |
88 ms |
34864 KB |
Output is correct |
| 24 |
Correct |
97 ms |
37468 KB |
Output is correct |
| 25 |
Correct |
143 ms |
40868 KB |
Output is correct |
| 26 |
Correct |
70 ms |
34984 KB |
Output is correct |
| 27 |
Correct |
83 ms |
34744 KB |
Output is correct |
| 28 |
Correct |
87 ms |
34952 KB |
Output is correct |
| 29 |
Correct |
95 ms |
36768 KB |
Output is correct |
| 30 |
Correct |
112 ms |
38644 KB |
Output is correct |
| 31 |
Correct |
94 ms |
35052 KB |
Output is correct |
| 32 |
Correct |
76 ms |
34840 KB |
Output is correct |
| 33 |
Correct |
81 ms |
35080 KB |
Output is correct |
| 34 |
Correct |
95 ms |
37444 KB |
Output is correct |
| 35 |
Correct |
127 ms |
40928 KB |
Output is correct |
| 36 |
Correct |
93 ms |
37440 KB |
Output is correct |
