結果
問題 | 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); }