結果
問題 | No.2337 Equidistant |
ユーザー | SSRS |
提出日時 | 2022-08-22 23:03:39 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,644 bytes |
コンパイル時間 | 1,662 ms |
コンパイル使用メモリ | 176,976 KB |
実行使用メモリ | 25,552 KB |
最終ジャッジ日時 | 2024-04-28 10:58:35 |
合計ジャッジ時間 | 14,468 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
5,248 KB |
testcase_01 | AC | 1 ms
5,376 KB |
testcase_02 | AC | 1 ms
5,376 KB |
testcase_03 | AC | 2 ms
5,376 KB |
testcase_04 | AC | 1 ms
5,376 KB |
testcase_05 | AC | 1 ms
5,376 KB |
testcase_06 | AC | 6 ms
5,376 KB |
testcase_07 | AC | 6 ms
5,376 KB |
testcase_08 | AC | 6 ms
5,376 KB |
testcase_09 | AC | 6 ms
5,376 KB |
testcase_10 | AC | 6 ms
5,376 KB |
testcase_11 | AC | 578 ms
25,388 KB |
testcase_12 | AC | 574 ms
25,552 KB |
testcase_13 | AC | 572 ms
25,444 KB |
testcase_14 | AC | 564 ms
25,392 KB |
testcase_15 | AC | 604 ms
25,388 KB |
testcase_16 | AC | 598 ms
25,416 KB |
testcase_17 | AC | 572 ms
25,404 KB |
testcase_18 | AC | 590 ms
25,504 KB |
testcase_19 | AC | 582 ms
25,412 KB |
testcase_20 | AC | 578 ms
25,404 KB |
testcase_21 | TLE | - |
testcase_22 | -- | - |
testcase_23 | -- | - |
testcase_24 | -- | - |
testcase_25 | -- | - |
testcase_26 | -- | - |
testcase_27 | -- | - |
testcase_28 | -- | - |
ソースコード
#include <bits/stdc++.h> using namespace std; int main(){ int N, Q; cin >> N >> Q; vector<vector<int>> E(N); for (int i = 0; i < N - 1; i++){ int U, V; cin >> U >> V; U--; V--; E[U].push_back(V); E[V].push_back(U); } vector<int> p(N, -1); vector<vector<int>> c(N); vector<int> d(N, 0); queue<int> q; q.push(0); vector<int> bfs; while (!q.empty()){ int v = q.front(); q.pop(); bfs.push_back(v); for (int w : E[v]){ if (w != p[v]){ p[w] = v; c[v].push_back(w); d[w] = d[v] + 1; q.push(w); } } } reverse(bfs.begin(), bfs.end()); vector<int> dp(N, 1); for (int v : bfs){ for (int w : c[v]){ dp[v] += dp[w]; } } auto naive_lca = [&](int u, int v){ while (d[u] > d[v]){ u = p[u]; } while (d[v] > d[u]){ v = p[v]; } while (u != v){ u = p[u]; v = p[v]; } return u; }; auto naive_la = [&](int v, int k){ for (int i = 0; i < k; i++){ v = p[v]; } return v; }; for (int i = 0; i < Q; i++){ int A, B; cin >> A >> B; A--; B--; if (d[A] > d[B]){ swap(A, B); } if (d[A] % 2 != d[B] % 2){ cout << 0 << endl; } else { int C = naive_lca(A, B); if (d[A] == d[B]){ int A2 = naive_la(A, d[A] - d[C] - 1); int B2 = naive_la(B, d[B] - d[C] - 1); cout << N - dp[A2] - dp[B2] << endl; } else { int D = d[A] + d[B] - d[C] * 2; int X = naive_la(B, D / 2); int Y = naive_la(B, D / 2 - 1); cout << dp[X] - dp[Y] << endl; } } } }