結果
問題 | No.1094 木登り / Climbing tree |
ユーザー | Akidai16 |
提出日時 | 2021-05-09 06:53:55 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 1,252 ms / 2,000 ms |
コード長 | 2,354 bytes |
コンパイル時間 | 4,447 ms |
コンパイル使用メモリ | 240,180 KB |
実行使用メモリ | 46,848 KB |
最終ジャッジ日時 | 2024-11-08 07:18:23 |
合計ジャッジ時間 | 30,896 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
5,248 KB |
testcase_01 | AC | 1,011 ms
38,784 KB |
testcase_02 | AC | 267 ms
46,848 KB |
testcase_03 | AC | 228 ms
5,248 KB |
testcase_04 | AC | 255 ms
18,432 KB |
testcase_05 | AC | 394 ms
33,920 KB |
testcase_06 | AC | 551 ms
14,464 KB |
testcase_07 | AC | 1,045 ms
38,656 KB |
testcase_08 | AC | 1,093 ms
38,656 KB |
testcase_09 | AC | 1,252 ms
38,784 KB |
testcase_10 | AC | 1,074 ms
38,784 KB |
testcase_11 | AC | 1,070 ms
38,784 KB |
testcase_12 | AC | 1,021 ms
38,656 KB |
testcase_13 | AC | 1,017 ms
38,784 KB |
testcase_14 | AC | 1,073 ms
38,656 KB |
testcase_15 | AC | 635 ms
12,800 KB |
testcase_16 | AC | 943 ms
36,608 KB |
testcase_17 | AC | 809 ms
22,912 KB |
testcase_18 | AC | 729 ms
17,920 KB |
testcase_19 | AC | 873 ms
30,976 KB |
testcase_20 | AC | 1,068 ms
38,656 KB |
testcase_21 | AC | 809 ms
24,448 KB |
testcase_22 | AC | 1,064 ms
38,784 KB |
testcase_23 | AC | 1,037 ms
38,656 KB |
testcase_24 | AC | 1,103 ms
38,784 KB |
testcase_25 | AC | 1,046 ms
38,784 KB |
testcase_26 | AC | 1,028 ms
38,784 KB |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using namespace atcoder; using ll = long long; constexpr ll mod = 1e9 + 7; constexpr ll INF = 1LL << 60; #define REP(i, init, n) for(int i = (int)(init); i < (int)(n); i++) #define vi vector<int> #define vl vector<long> #define vvi vector<vector<int>> #define vvl vector<vector<long>> #define pint pair<int, int> #define plong pair<long, long> int N, Q, digit; vector<vector<pint>> G; vvi Parents; vl dist, depth; template<class T> void vv_output(vector<vector<T>> A, bool blank = true) { for(auto row: A) { for(auto a: row) { if(blank) cout << a << ' '; else cout << a; } cout << endl; } } void dfs(int parent) { for(auto child: G[parent]) { if(child.first == 0 || Parents[child.first][0] >= 0) continue; Parents[child.first][0] = parent; dist[child.first] = dist[parent] + child.second; depth[child.first] = depth[parent] + 1; dfs(child.first); } } int lcs(int u, int v) { if(depth[u] > depth[v]) swap(u, v); for(int k = 0; k < digit; k++) { if((depth[v] - depth[u]) >> k & 1) { v = Parents[v][k]; } } if(u == v) return u; for(int k = digit - 1; k >= 0; k--) { if(Parents[u][k] != Parents[v][k]) { u = Parents[u][k]; v = Parents[v][k]; } } return Parents[u][0]; } void solve() { digit = 1; int n = N; while(n > 1) { n >>= 1; digit++; } Parents.resize(N, vi(digit, -1)); dist.resize(N, 0); depth.resize(N, 0); dfs(0); // vv_output(Parents); REP(i, 1, digit) { REP(j, 0, N) { if(Parents[j][i - 1] == -1) continue; Parents[j][i] = Parents[Parents[j][i - 1]][i - 1]; } } // vv_output(Parents); cin >> Q; REP(i, 0, Q) { int s, t; cin >> s >> t; s--; t--; int parent = lcs(s, t); // cout << s << ' ' << t << ' ' << parent << endl; cout << dist[s] + dist[t] - 2 * dist[parent] << endl; } } int main() { cin >> N; G.resize(N); REP(i, 0, N - 1) { int a, b, c; cin >> a >> b >> c; a--; b--; G[a].push_back({b, c}); G[b].push_back({a, c}); } solve(); }