結果
問題 | No.2337 Equidistant |
ユーザー | SSRS |
提出日時 | 2023-06-02 20:18:36 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 390 ms / 4,000 ms |
コード長 | 2,881 bytes |
コンパイル時間 | 1,763 ms |
コンパイル使用メモリ | 176,892 KB |
実行使用メモリ | 42,152 KB |
最終ジャッジ日時 | 2023-08-28 02:27:14 |
合計ジャッジ時間 | 10,160 ms |
ジャッジサーバーID (参考情報) |
judge12 / judge14 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 1 ms
4,380 KB |
testcase_01 | AC | 1 ms
4,380 KB |
testcase_02 | AC | 1 ms
4,380 KB |
testcase_03 | AC | 1 ms
4,376 KB |
testcase_04 | AC | 1 ms
4,384 KB |
testcase_05 | AC | 2 ms
4,376 KB |
testcase_06 | AC | 3 ms
4,380 KB |
testcase_07 | AC | 3 ms
4,380 KB |
testcase_08 | AC | 3 ms
4,376 KB |
testcase_09 | AC | 3 ms
4,376 KB |
testcase_10 | AC | 3 ms
4,380 KB |
testcase_11 | AC | 227 ms
39,324 KB |
testcase_12 | AC | 227 ms
39,324 KB |
testcase_13 | AC | 224 ms
39,444 KB |
testcase_14 | AC | 225 ms
39,256 KB |
testcase_15 | AC | 226 ms
39,300 KB |
testcase_16 | AC | 227 ms
39,260 KB |
testcase_17 | AC | 227 ms
39,284 KB |
testcase_18 | AC | 227 ms
39,280 KB |
testcase_19 | AC | 229 ms
39,524 KB |
testcase_20 | AC | 234 ms
39,408 KB |
testcase_21 | AC | 233 ms
42,096 KB |
testcase_22 | AC | 210 ms
38,432 KB |
testcase_23 | AC | 200 ms
38,672 KB |
testcase_24 | AC | 293 ms
42,152 KB |
testcase_25 | AC | 210 ms
38,488 KB |
testcase_26 | AC | 390 ms
42,048 KB |
testcase_27 | AC | 251 ms
38,528 KB |
testcase_28 | AC | 245 ms
38,360 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 s = la(S, d[S] - d[V] - 1); int t = la(T, d[T] - d[V] - 1); cout << N - dp[s] - dp[t] << '\n'; } else { int D = d[S] + d[T] - d[V] * 2; int s = la(T, D / 2); int t = la(T, D / 2 - 1); cout << dp[s] - dp[t] << '\n'; } } } }