結果

問題 No.439 チワワのなる木
ユーザー femtofemto
提出日時 2016-10-28 23:12:32
言語 C++14
(gcc 12.3.0 + boost 1.83.0)
結果
AC  
実行時間 48 ms / 5,000 ms
コード長 924 bytes
コンパイル時間 1,344 ms
コンパイル使用メモリ 171,524 KB
実行使用メモリ 16,772 KB
最終ジャッジ日時 2024-11-24 06:19:56
合計ジャッジ時間 2,579 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge4
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 AC 3 ms
5,760 KB
testcase_01 AC 3 ms
5,760 KB
testcase_02 AC 3 ms
5,760 KB
testcase_03 AC 2 ms
5,760 KB
testcase_04 AC 2 ms
5,760 KB
testcase_05 AC 2 ms
5,888 KB
testcase_06 AC 3 ms
6,016 KB
testcase_07 AC 2 ms
6,016 KB
testcase_08 AC 2 ms
5,888 KB
testcase_09 AC 2 ms
5,888 KB
testcase_10 AC 3 ms
5,888 KB
testcase_11 AC 3 ms
5,888 KB
testcase_12 AC 2 ms
5,888 KB
testcase_13 AC 3 ms
5,888 KB
testcase_14 AC 3 ms
5,760 KB
testcase_15 AC 3 ms
5,760 KB
testcase_16 AC 4 ms
5,888 KB
testcase_17 AC 3 ms
5,888 KB
testcase_18 AC 32 ms
9,472 KB
testcase_19 AC 27 ms
9,216 KB
testcase_20 AC 40 ms
10,440 KB
testcase_21 AC 12 ms
7,296 KB
testcase_22 AC 13 ms
7,040 KB
testcase_23 AC 43 ms
11,264 KB
testcase_24 AC 48 ms
15,712 KB
testcase_25 AC 31 ms
10,220 KB
testcase_26 AC 31 ms
10,776 KB
testcase_27 AC 36 ms
16,772 KB
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
typedef long long ll;

int N;
ll ans;
ll C[100000];
ll W[100000];
vector<int> G[100000];
string S;

void dfs1(int n, int p) {
	if(S[n] == 'c') C[n]++;
	else W[n]++;
	for(auto c : G[n]) {
		if(c == p) continue;
		dfs1(c, n);
		C[n] += C[c];
		W[n] += W[c];
	}
}

void dfs2(int n, int p) {
	if(S[n] == 'w') {
		ll csum = C[0], wsum = W[0];
		for(auto c : G[n]) {
			ll cn = 0, wn = 0;
			if(c == p) {
				cn = csum - C[n];
				wn = wsum - W[n];
			}
			else {
				cn = C[c];
				wn = W[c];
			}
			ans += cn * (wsum - 1 - wn);
			//ans += wn * (csum - cn);
		}
	}
	for(auto c : G[n]) {
		if(c == p) continue;
		dfs2(c, n);
	}
}

int main() {
	cin.tie(0);
	ios::sync_with_stdio(false);

	cin >> N;
	cin >> S;
	for(int i = 0; i < N - 1; i++) {
		int a, b;
		cin >> a >> b;
		a--, b--;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	dfs1(0, -1);
	dfs2(0, -1);

	cout << ans << endl;
}
0