結果
問題 | No.1418 Sum of Sum of Subtree Size |
ユーザー |
![]() |
提出日時 | 2021-03-05 21:47:46 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 67 ms / 2,000 ms |
コード長 | 722 bytes |
コンパイル時間 | 382 ms |
コンパイル使用メモリ | 49,280 KB |
実行使用メモリ | 16,640 KB |
最終ジャッジ日時 | 2024-10-07 01:22:43 |
合計ジャッジ時間 | 2,717 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 41 |
ソースコード
#include <stdio.h>#include <vector>int a;long long int ans = 0;std::vector<int> V[100010];int check[100010];int getSize(int k, int prev){if(check[k]) return check[k];int ans = 1;for(int i=0;i<V[k].size();i++){if(V[k][i]==prev) continue;ans += getSize(V[k][i],k);}return check[k] = ans;}void func(int k, int prev){ans += 1LL * (a-getSize(k,prev))*getSize(k,prev);ans += a;for(int i=0;i<V[k].size();i++){if(V[k][i]==prev) continue;int t = getSize(V[k][i],k);ans += 1LL * (a-t) * t;func(V[k][i],k);}}int main(){scanf("%d",&a);for(int i=1;i<a;i++){int b,c;scanf("%d%d",&b,&c);V[b].push_back(c);V[c].push_back(b);}func(1,0);printf("%lld\n",ans);}