結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー publflpublfl
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #
プレゼンテーションモードにする

#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);
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0