結果

問題 No.439 チワワのなる木
ユーザー vjudge1vjudge1
提出日時 2024-12-22 11:14:23
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 61 ms / 5,000 ms
コード長 758 bytes
コンパイル時間 2,030 ms
コンパイル使用メモリ 168,976 KB
実行使用メモリ 17,136 KB
最終ジャッジ日時 2024-12-22 11:14:27
合計ジャッジ時間 3,862 ms
ジャッジサーバーID
（参考情報） 		judge2 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 4 ms
6,144 KB
testcase_01 AC 3 ms
6,016 KB
testcase_02 AC 3 ms
6,016 KB
testcase_03 AC 3 ms
6,016 KB
testcase_04 AC 3 ms
6,272 KB
testcase_05 AC 3 ms
6,016 KB
testcase_06 AC 3 ms
6,144 KB
testcase_07 AC 2 ms
6,016 KB
testcase_08 AC 3 ms
6,144 KB
testcase_09 AC 3 ms
5,888 KB
testcase_10 AC 3 ms
6,144 KB
testcase_11 AC 3 ms
6,144 KB
testcase_12 AC 3 ms
6,016 KB
testcase_13 AC 3 ms
6,144 KB
testcase_14 AC 3 ms
6,016 KB
testcase_15 AC 4 ms
6,144 KB
testcase_16 AC 4 ms
6,144 KB
testcase_17 AC 3 ms
6,016 KB
testcase_18 AC 39 ms
9,728 KB
testcase_19 AC 33 ms
9,704 KB
testcase_20 AC 47 ms
11,008 KB
testcase_21 AC 14 ms
7,680 KB
testcase_22 AC 14 ms
7,424 KB
testcase_23 AC 49 ms
11,900 KB
testcase_24 AC 61 ms
16,256 KB
testcase_25 AC 45 ms
10,636 KB
testcase_26 AC 40 ms
11,392 KB
testcase_27 AC 42 ms
17,136 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include<bits/stdc++.h>
#define FL(i,a,b) for(ll i=(a);i<=(b);i++)
#define FR(i,a,b) for(ll i=(a);i>=(b);i--)
#define ll long long
using namespace std;
const ll MAXN = 1e5 + 10;
ll n,sumc=0,sumw=0,ans=0;
ll f[MAXN],g[MAXN];
vector<ll>G[MAXN];
char s[MAXN];
void dfs(ll u,ll fa){
	if(s[u]=='w') f[u]++;
	else g[u]++;
	for(ll v:G[u]){
		if(v==fa) continue;
		dfs(v,u);
		f[u]+=f[v];
		g[u]+=g[v];
	}
	if(s[u]=='w'){
		for(ll v:G[u]){
			if(v==fa) continue;
			ans+=g[v]*(sumw-f[v]-1);
		}
		ans+=(sumc-g[u])*(f[u]-1);
	}
}
signed main(){
	scanf("%lld",&n);
	scanf("%s",s+1);
	FL(i,1,n-1){
		ll u,v;
		scanf("%lld%lld",&u,&v);
		G[u].push_back(v);
		G[v].push_back(u);
	}
	FL(i,1,n){
		if(s[i]=='w') sumw++;
		else sumc++;
	}
	dfs(1,0);
	printf("%lld\n",ans);
}
0