結果
問題 | No.399 動的な領主 |
ユーザー | legendcn666 |
提出日時 | 2024-10-13 00:20:35 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,430 bytes |
コンパイル時間 | 1,729 ms |
コンパイル使用メモリ | 171,612 KB |
実行使用メモリ | 42,840 KB |
最終ジャッジ日時 | 2024-10-13 00:20:40 |
合計ジャッジ時間 | 5,112 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 4 ms
9,244 KB |
testcase_01 | AC | 3 ms
6,816 KB |
testcase_02 | WA | - |
testcase_03 | WA | - |
testcase_04 | WA | - |
testcase_05 | WA | - |
testcase_06 | WA | - |
testcase_07 | WA | - |
testcase_08 | WA | - |
testcase_09 | WA | - |
testcase_10 | WA | - |
testcase_11 | WA | - |
testcase_12 | WA | - |
testcase_13 | WA | - |
testcase_14 | WA | - |
testcase_15 | WA | - |
testcase_16 | WA | - |
testcase_17 | WA | - |
testcase_18 | WA | - |
ソースコード
# include <bits/stdc++.h> using namespace std; typedef long long ll; # define int long long # define lc u << 1 # define rc u << 1 | 1 # define fi first # define se second const int N = 100005; int n, q; vector <int> e[N]; int f[N][35], dep[N]; void dfs (int u, int fa) { f[u][0] = fa, dep[u] = dep[fa] + 1; for (int v : e[u]) { if (v == fa) continue; dfs (v, u); } } int getlca (int u, int v) { if (dep[u] < dep[v]) swap (u, v); for (int i = 30; i >= 0; i -- ) { if (dep[v] - dep[u] >= 1 << i) u = f[u][i]; } if (u == v) return u; for (int i = 30; i >= 0; i -- ) { if (f[u][i] != f[v][i]) u = f[u][i], v = f[v][i]; } return f[u][0]; } int cnt[N]; int ans; void dfs2 (int u, int fa) { for (int v : e[u]) { if (v == fa) continue; dfs2 (v, u); } cnt[fa] += cnt[u]; ans += cnt[u] * (cnt[u] + 1) / 2; } signed main () { // freopen ("tax.in", "r", stdin); freopen ("tax.out", "w", stdout); scanf ("%lld", &n); for (int i = 1; i < n; i ++ ) { int u, v; scanf ("%lld%lld", &u, &v); u --, v -- ; e[u].push_back (v), e[v].push_back (u); } dfs (0, -1); for (int j = 1; j <= 30; j ++ ) { for (int i = 0; i < n; i ++ ) f[i][j] = f[f[i][j - 1]][j - 1]; } scanf ("%lld", &q); while (q -- ) { int u, v; scanf ("%lld%lld", &u, &v); u --, v -- ; int lca = getlca (u, v); cnt[u] ++ , cnt[v] ++ , cnt[lca] -- , cnt[f[lca][0]] -- ; } dfs2 (0, -1); printf ("%lld\n", ans); return 0; }