結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 21:40:12 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 100 ms / 2,000 ms |
コード長 | 944 bytes |
コンパイル時間 | 1,852 ms |
コンパイル使用メモリ | 176,972 KB |
実行使用メモリ | 16,400 KB |
最終ジャッジ日時 | 2024-10-07 01:06:38 |
合計ジャッジ時間 | 4,394 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <bits/stdc++.h> using namespace std; int main(){ int N; cin >> N; 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> b; queue<int> Q; Q.push(0); while (!Q.empty()){ int v = Q.front(); Q.pop(); b.push_back(v); for (int w : E[v]){ if (w != p[v]){ p[w] = v; c[v].push_back(w); Q.push(w); } } } reverse(b.begin(), b.end()); vector<long long> dp1(N, 1), dp2(N, 0); long long ans = 0; for (int v : b){ int d = c[v].size(); for (int w : c[v]){ dp1[v] += dp1[w]; } for (int w : c[v]){ ans += (dp2[w] + dp1[w]) * (dp1[v] - dp1[w]) * 2; } for (int w : c[v]){ dp2[v] += dp2[w] + dp1[w]; } } ans += (long long) N * N; cout << ans << endl; }