結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー | tnakao0123 |
提出日時 | 2021-03-13 03:21:19 |
言語 | C++14 (gcc 12.3.0 + boost 1.83.0) |
結果 |
AC
|
実行時間 | 56 ms / 2,000 ms |
コード長 | 1,650 bytes |
コンパイル時間 | 884 ms |
コンパイル使用メモリ | 95,872 KB |
実行使用メモリ | 12,160 KB |
最終ジャッジ日時 | 2024-04-22 16:24:43 |
合計ジャッジ時間 | 3,913 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge2 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 3 ms
6,400 KB |
testcase_01 | AC | 4 ms
6,400 KB |
testcase_02 | AC | 3 ms
6,400 KB |
testcase_03 | AC | 52 ms
12,032 KB |
testcase_04 | AC | 56 ms
11,904 KB |
testcase_05 | AC | 51 ms
12,032 KB |
testcase_06 | AC | 49 ms
12,160 KB |
testcase_07 | AC | 54 ms
11,904 KB |
testcase_08 | AC | 35 ms
10,240 KB |
testcase_09 | AC | 18 ms
8,192 KB |
testcase_10 | AC | 19 ms
8,320 KB |
testcase_11 | AC | 12 ms
7,680 KB |
testcase_12 | AC | 27 ms
9,472 KB |
testcase_13 | AC | 32 ms
9,984 KB |
testcase_14 | AC | 33 ms
9,984 KB |
testcase_15 | AC | 28 ms
9,216 KB |
testcase_16 | AC | 6 ms
6,912 KB |
testcase_17 | AC | 6 ms
6,656 KB |
testcase_18 | AC | 48 ms
11,520 KB |
testcase_19 | AC | 11 ms
7,296 KB |
testcase_20 | AC | 5 ms
6,528 KB |
testcase_21 | AC | 26 ms
9,088 KB |
testcase_22 | AC | 23 ms
8,704 KB |
testcase_23 | AC | 4 ms
6,528 KB |
testcase_24 | AC | 5 ms
6,656 KB |
testcase_25 | AC | 4 ms
6,528 KB |
testcase_26 | AC | 4 ms
6,528 KB |
testcase_27 | AC | 3 ms
6,528 KB |
testcase_28 | AC | 3 ms
6,528 KB |
testcase_29 | AC | 5 ms
6,784 KB |
testcase_30 | AC | 5 ms
6,656 KB |
testcase_31 | AC | 4 ms
6,528 KB |
testcase_32 | AC | 5 ms
6,656 KB |
testcase_33 | AC | 11 ms
7,680 KB |
testcase_34 | AC | 47 ms
11,904 KB |
testcase_35 | AC | 19 ms
8,704 KB |
testcase_36 | AC | 7 ms
7,168 KB |
testcase_37 | AC | 33 ms
11,792 KB |
testcase_38 | AC | 32 ms
11,148 KB |
testcase_39 | AC | 3 ms
6,528 KB |
testcase_40 | AC | 3 ms
6,528 KB |
testcase_41 | AC | 3 ms
6,528 KB |
testcase_42 | AC | 3 ms
6,400 KB |
testcase_43 | AC | 4 ms
6,400 KB |
ソースコード
/* -*- coding: utf-8 -*- * * 1418.cc: No.1418 Sum of Sum of Subtree Size - yukicoder */ #include<cstdio> #include<cstdlib> #include<cstring> #include<cmath> #include<iostream> #include<string> #include<vector> #include<map> #include<set> #include<stack> #include<list> #include<queue> #include<deque> #include<algorithm> #include<numeric> #include<utility> #include<complex> #include<functional> using namespace std; /* constant */ const int MAX_N = 100000; /* typedef */ typedef long long ll; typedef vector<int> vi; /* global variables */ vi nbrs[MAX_N]; int ps[MAX_N], cis[MAX_N]; int szs[MAX_N]; ll ss[MAX_N], rss[MAX_N]; /* subroutines */ /* main */ int main() { int n; scanf("%d", &n); for (int i = 1; i < n; i++) { int u, v; scanf("%d%d", &u, &v); u--, v--; nbrs[u].push_back(v); nbrs[v].push_back(u); } ps[0] = -1; for (int u = 0; u >= 0;) { vi &nbru = nbrs[u]; if (cis[u] < nbru.size() && nbru[cis[u]] == ps[u]) cis[u]++; if (cis[u] < nbru.size()) { int v = nbru[cis[u]++]; ps[v] = u; u = v; } else { szs[u]++; ss[u] += szs[u]; if (ps[u] >= 0) { szs[ps[u]] += szs[u]; ss[ps[u]] += ss[u]; } u = ps[u]; } } rss[0] = ss[0]; ll sum = rss[0]; memset(cis, 0, sizeof(cis)); for (int u = 0; u >= 0;) { vi &nbru = nbrs[u]; if (cis[u] < nbru.size() && nbru[cis[u]] == ps[u]) cis[u]++; if (cis[u] < nbru.size()) { int v = nbru[cis[u]++]; rss[v] = rss[u] + n - szs[v] * 2; sum += rss[v]; u = v; } else { u = ps[u]; } } printf("%lld\n", sum); return 0; }