結果
問題 | No.399 動的な領主 |
ユーザー | pew pew pew |
提出日時 | 2024-10-13 00:39:07 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 114 ms / 2,000 ms |
コード長 | 1,340 bytes |
コンパイル時間 | 1,445 ms |
コンパイル使用メモリ | 168,808 KB |
実行使用メモリ | 30,960 KB |
最終ジャッジ日時 | 2024-10-13 00:39:12 |
合計ジャッジ時間 | 3,270 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 3 ms
5,248 KB |
testcase_01 | AC | 2 ms
5,248 KB |
testcase_02 | AC | 2 ms
5,248 KB |
testcase_03 | AC | 2 ms
5,248 KB |
testcase_04 | AC | 3 ms
5,248 KB |
testcase_05 | AC | 10 ms
5,888 KB |
testcase_06 | AC | 97 ms
26,240 KB |
testcase_07 | AC | 90 ms
26,240 KB |
testcase_08 | AC | 84 ms
26,112 KB |
testcase_09 | AC | 82 ms
26,204 KB |
testcase_10 | AC | 3 ms
5,248 KB |
testcase_11 | AC | 8 ms
6,016 KB |
testcase_12 | AC | 74 ms
26,240 KB |
testcase_13 | AC | 75 ms
26,368 KB |
testcase_14 | AC | 59 ms
30,960 KB |
testcase_15 | AC | 64 ms
30,880 KB |
testcase_16 | AC | 64 ms
28,160 KB |
testcase_17 | AC | 114 ms
26,276 KB |
testcase_18 | AC | 89 ms
26,240 KB |
ソースコード
# include <bits/stdc++.h> using namespace std; typedef long long ll; const int N = 100010; struct edge {ll to, next;} e[N << 1]; ll n, q, ans; ll cnt, head[N]; ll dep[N], d[N]; ll f[N][22]; void add(ll u, ll v) {e[++cnt] = {v, head[u]}, head[u] = cnt;} void DFS(ll u, ll fa) { dep[u] = dep[fa] + 1; f[u][0] = fa; for (int i = 1; (1 << i) <= dep[u]; i++) f[u][i] = f[f[u][i - 1]][i - 1]; ll v; for (int i = head[u]; i; i = e[i].next) { v = e[i].to; if (v == fa) continue; DFS(v, u); } } ll DFS2(ll u, ll fa) { ll v; for (int i = head[u]; i; i = e[i].next) { v = e[i].to; if (v == fa) continue; d[u] += DFS2(v, u); } ans += (d[u] + 1) * d[u] / 2; return d[u]; } ll LCA(ll u, ll v) { if (dep[u] < dep[v]) swap(u, v); ll txp; txp = dep[u] - dep[v]; for (int i = 21; i >= 0; i--) if (txp & (1 << i)) u = f[u][i]; if (u == v) return u; for (int i = 21; i >= 0; i--) if (f[u][i] != f[v][i]) u = f[u][i], v = f[v][i]; return f[u][0]; } int main() { scanf("%lld", &n); for (int i = 1; i < n; i++) { ll u, v; scanf("%lld%lld", &u, &v); add(u, v), add(v, u); } DFS(1, 0); scanf("%lld", &q); while (q--) { ll a, b, lca; scanf("%lld%lld", &a, &b); lca = LCA(a, b); d[a]++, d[b]++; d[lca]--; if (f[lca][0] >= 0) d[f[lca][0]]--; } ans = 0; DFS2(1, 0); printf("%lld\n", ans); return 0; }