# | Time | Username | Problem | Language | Result | Execution time | Memory |
---|---|---|---|---|---|---|---|
1199233 | gugug | Evacuation plan (IZhO18_plan) | C++17 | 4093 ms | 28436 KiB |
#pragma GCC optimize("Ofast,unroll-loops")
#include <bits/stdc++.h>
using namespace std;
#define pii pair<int, int>
#define iii tuple<int, int, int>
int n, m, k, q, parent[100001], sz[100001], ans[100001];
iii queries[100000];
vector<pii> adj[100001];
vector<int> aadj[100001];
pii dist[100001];
bool flag[100001];
priority_queue<pii, vector<pii>, greater<pii>> pq;
int find(int x) {
if (x == parent[x]) return x;
return parent[x] = find(parent[x]);
}
void unite(int u, int v) {
u = find(u), v = find(v);
if (u == v) return;
if (sz[u] > sz[v]) {
sz[u] += sz[v];
parent[v] = u;
} else {
sz[v] += sz[u];
parent[u] = v;
}
}
int main() {
scanf("%d%d", &n, &m);
for (int i = 1; i <= n; i++) {
parent[i] = i;
sz[i] = 1;
dist[i] = {0x3f3f3f3f, i};
}
for (int i = 1; i <= m; i++) {
int u, v, w; scanf("%d %d %d", &u, &v, &w);
adj[u].push_back({v, w});
adj[v].push_back({u, w});
}
scanf("%d", &k);
for (int i = 0; i < k; i++) {
int g; scanf("%d", &g);
pq.push({0, g}); dist[g].first = 0;
}
while (!pq.empty()) {
auto [d, u] = pq.top(); pq.pop();
if (flag[u]) continue;
flag[u] = 1;
for (auto [v, w]: adj[u]) if (!flag[v] && dist[v].first > dist[u].first + w) {
dist[v].first = dist[u].first + w;
pq.push({dist[v].first, v});
}
}
sort(dist + 1, dist + n + 1, greater<pii>());
scanf("%d", &q);
for (int i = 0; i < q; i++) {
auto& [a, b, c] = queries[i];
scanf("%d %d", &a, &b);
c = i;
}
for (int i = 1; i <= n; i++) {
auto [d, u] = dist[i];
for (auto [v, w]: adj[u]) aadj[v].push_back(u);
for (auto v: aadj[u]) unite(u, v);
for (int j = 0; j < q; j++) {
auto [a, b, c] = queries[j];
if (find(a) == find(b) && !ans[c]) ans[c] = d;
}
}
for (int i = 0; i < q; i++) printf("%d\n", ans[i]);
}
Compilation message (stderr)
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |
# | Verdict | Execution time | Memory | Grader output |
---|---|---|---|---|
Fetching results... |