結果
問題 | No.2337 Equidistant |
ユーザー | SSRS |
提出日時 | 2023-06-02 20:17:27 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 675 ms / 4,000 ms |
コード長 | 2,885 bytes |
コンパイル時間 | 1,933 ms |
コンパイル使用メモリ | 177,664 KB |
実行使用メモリ | 42,184 KB |
最終ジャッジ日時 | 2023-08-28 02:27:03 |
合計ジャッジ時間 | 10,635 ms |
ジャッジサーバーID (参考情報) |
judge15 / judge12 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 2 ms
4,376 KB |
testcase_01 | AC | 2 ms
4,376 KB |
testcase_02 | AC | 1 ms
4,380 KB |
testcase_03 | AC | 1 ms
4,380 KB |
testcase_04 | AC | 1 ms
4,380 KB |
testcase_05 | AC | 1 ms
4,376 KB |
testcase_06 | AC | 3 ms
4,376 KB |
testcase_07 | AC | 3 ms
4,380 KB |
testcase_08 | AC | 3 ms
4,376 KB |
testcase_09 | AC | 3 ms
4,380 KB |
testcase_10 | AC | 3 ms
4,376 KB |
testcase_11 | AC | 296 ms
39,224 KB |
testcase_12 | AC | 304 ms
39,272 KB |
testcase_13 | AC | 302 ms
39,292 KB |
testcase_14 | AC | 299 ms
39,244 KB |
testcase_15 | AC | 297 ms
39,392 KB |
testcase_16 | AC | 314 ms
39,276 KB |
testcase_17 | AC | 301 ms
39,228 KB |
testcase_18 | AC | 297 ms
39,232 KB |
testcase_19 | AC | 306 ms
39,324 KB |
testcase_20 | AC | 302 ms
39,292 KB |
testcase_21 | AC | 363 ms
42,184 KB |
testcase_22 | AC | 257 ms
38,404 KB |
testcase_23 | AC | 251 ms
38,584 KB |
testcase_24 | AC | 503 ms
42,108 KB |
testcase_25 | AC | 269 ms
38,460 KB |
testcase_26 | AC | 675 ms
42,100 KB |
testcase_27 | AC | 309 ms
38,436 KB |
testcase_28 | AC | 296 ms
38,384 KB |
ソースコード
#include <bits/stdc++.h> using namespace std; const int LOG = 17; struct lowest_common_ancestor{ vector<int> d; vector<vector<int>> pp; lowest_common_ancestor(vector<int> &p, vector<int> &d): d(d){ int N = p.size(); pp = vector<vector<int>>(LOG, vector<int>(N, -1)); pp[0] = p; for (int i = 0; i < LOG - 1; i++){ for (int j = 0; j < N; j++){ if (pp[i][j] != -1){ pp[i + 1][j] = pp[i][pp[i][j]]; } } } } int la(int v, int x){ for (int i = 0; i < LOG; i++){ if ((x >> i & 1) == 1){ v = pp[i][v]; } } return v; } int lca(int u, int v){ if (d[u] > d[v]){ swap(u, v); } v = la(v, d[v] - d[u]); if (u == v){ return u; } for (int i = LOG - 1; i >= 0; i--){ if (pp[i][u] != pp[i][v]){ u = pp[i][u]; v = pp[i][v]; } } return pp[0][u]; } }; int main(){ ios_base::sync_with_stdio(false); cin.tie(nullptr); int N, Q; cin >> N >> Q; vector<vector<int>> E(N); for (int i = 0; i < N - 1; i++){ int A, B; cin >> A >> B; A--; B--; E[A].push_back(B); E[B].push_back(A); } 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]; } } vector<vector<int>> pp(LOG, vector<int>(N, -1)); pp[0] = p; for (int i = 0; i < LOG - 1; i++){ for (int j = 0; j < N; j++){ if (pp[i][j] != -1){ pp[i + 1][j] = pp[i][pp[i][j]]; } } } auto la = [&](int v, int x){ for (int i = 0; i < LOG; i++){ if ((x >> i & 1) == 1){ v = pp[i][v]; } } return v; }; auto lca = [&](int u, int v){ if (d[u] > d[v]){ swap(u, v); } v = la(v, d[v] - d[u]); if (u == v){ return u; } for (int i = LOG - 1; i >= 0; i--){ if (pp[i][u] != pp[i][v]){ u = pp[i][u]; v = pp[i][v]; } } return pp[0][u]; }; for (int i = 0; i < Q; i++){ int S, T; cin >> S >> T; S--; T--; if (d[S] > d[T]){ swap(S, T); } if (d[S] % 2 != d[T] % 2){ cout << 0 << '\n'; } else { int V = lca(S, T); if (d[S] == d[T]){ int S2 = la(S, d[S] - d[V] - 1); int T2 = la(T, d[T] - d[V] - 1); cout << N - dp[S2] - dp[T2] << '\n'; } else { int D = d[S] + d[T] - d[V] * 2; int U = la(T, D / 2); int V = la(T, D / 2 - 1); cout << dp[U] - dp[V] << '\n'; } } } }