結果
問題 | No.898 tri-βutree |
ユーザー | 0w1 |
提出日時 | 2019-10-10 17:50:17 |
言語 | C++17 (gcc 13.2.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 358 ms / 4,000 ms |
コード長 | 1,722 bytes |
コンパイル時間 | 2,774 ms |
コンパイル使用メモリ | 215,796 KB |
実行使用メモリ | 39,148 KB |
最終ジャッジ日時 | 2023-08-08 16:25:39 |
合計ジャッジ時間 | 10,695 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge11 |
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 235 ms
39,148 KB |
testcase_01 | AC | 1 ms
4,384 KB |
testcase_02 | AC | 2 ms
4,384 KB |
testcase_03 | AC | 2 ms
4,384 KB |
testcase_04 | AC | 2 ms
4,384 KB |
testcase_05 | AC | 2 ms
4,380 KB |
testcase_06 | AC | 2 ms
4,384 KB |
testcase_07 | AC | 348 ms
31,188 KB |
testcase_08 | AC | 354 ms
31,376 KB |
testcase_09 | AC | 348 ms
31,104 KB |
testcase_10 | AC | 352 ms
31,064 KB |
testcase_11 | AC | 335 ms
31,128 KB |
testcase_12 | AC | 346 ms
30,972 KB |
testcase_13 | AC | 338 ms
31,104 KB |
testcase_14 | AC | 338 ms
31,044 KB |
testcase_15 | AC | 337 ms
31,032 KB |
testcase_16 | AC | 338 ms
31,120 KB |
testcase_17 | AC | 338 ms
30,968 KB |
testcase_18 | AC | 345 ms
31,044 KB |
testcase_19 | AC | 346 ms
31,180 KB |
testcase_20 | AC | 358 ms
31,180 KB |
testcase_21 | AC | 343 ms
31,276 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; int main() { ios::sync_with_stdio(false); int N; cin >> N; vector<vector<pair<int, int>>> G(N); for (int i = 0; i < N - 1; ++i) { int U, V, W; cin >> U >> V >> W; G[U].emplace_back(W, V); G[V].emplace_back(W, U); } vector<int> dpt(N); vector<int64_t> d2r(N); vector<vector<int>> fa2(17, vector<int>(N, -1)); vector<vector<int64_t>> du2(17, vector<int64_t>(N)); { function<void(int, int)> dfs = [&](int u, int fa) { for (int i = 0; i < G[u].size(); ++i) { int w, v; tie(w, v) = G[u][i]; if (v == fa) continue; dpt[v] = dpt[u] + 1; d2r[v] = d2r[u] + w; fa2[0][v] = u; du2[0][v] = w; dfs(v, u); } }; dfs(0, -1); for (int i = 0; i + 1 < fa2.size(); ++i) { for (int u = 0; u < N; ++u) { int x = fa2[i][u]; if (x == -1) continue; fa2[i + 1][u] = fa2[i][x]; du2[i + 1][u] = du2[i][u] + du2[i][x]; } } } auto lca = [&](int u, int v) { if (dpt[u] < dpt[v]) swap(u, v); for (int i = fa2.size() - 1; dpt[u] != dpt[v]; --i) { if (dpt[u] - dpt[v] >= 1 << i) { u = fa2[i][u]; } } if (u == v) return u; for (int i = fa2.size() - 1; fa2[0][u] != fa2[0][v]; --i) { if (fa2[i][u] != fa2[i][v]) { u = fa2[i][u]; v = fa2[i][v]; } } assert(fa2[0][u] == fa2[0][v]); return fa2[0][u]; }; auto dis = [&](int u, int v) { return d2r[u] + d2r[v] - 2 * d2r[lca(u, v)]; }; int Q; cin >> Q; while (Q--) { int X, Y, Z; cin >> X >> Y >> Z; cout << (dis(X, Y) + dis(Y, Z) + dis(Z, X) >> 1) << endl; } return 0; }