結果

問題 No.1418 Sum of Sum of Subtree Size
ユーザー publfl
提出日時 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);
}
0