#127280 (C++11) No.439 チワワのなる木

• 問題: No.439 チワワのなる木
• ユーザー: hirokazu1020
• 提出日時: 2016-10-29 01:53:08
• 言語: C++11 (gcc 11.4.0)
• 結果: AC
• 実行時間: 73 ms / 5,000 ms
• コード長: 1,379 bytes
• コンパイル時間: 1,025 ms
• コンパイル使用メモリ: 98,220 KB
• 実行使用メモリ: 18,944 KB
• 最終ジャッジ日時: 2024-11-24 07:05:22
• 合計ジャッジ時間: 2,272 ms
• ジャッジサーバーID （参考情報）: judge2 / judge4

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	3 ms; 5,760 KB
testcase_01	AC	4 ms; 5,888 KB
testcase_02	AC	4 ms; 5,760 KB
testcase_03	AC	3 ms; 5,760 KB
testcase_04	AC	3 ms; 5,760 KB
testcase_05	AC	4 ms; 5,632 KB
testcase_06	AC	4 ms; 5,760 KB
testcase_07	AC	3 ms; 5,760 KB
testcase_08	AC	4 ms; 5,760 KB
testcase_09	AC	3 ms; 5,760 KB
testcase_10	AC	4 ms; 5,760 KB
testcase_11	AC	4 ms; 5,760 KB
testcase_12	AC	4 ms; 5,760 KB
testcase_13	AC	4 ms; 5,760 KB
testcase_14	AC	4 ms; 5,760 KB
testcase_15	AC	4 ms; 5,760 KB
testcase_16	AC	5 ms; 5,760 KB
testcase_17	AC	4 ms; 5,760 KB
testcase_18	AC	46 ms; 8,448 KB
testcase_19	AC	35 ms; 8,448 KB
testcase_20	AC	53 ms; 9,344 KB
testcase_21	AC	15 ms; 6,912 KB
testcase_22	AC	14 ms; 6,656 KB
testcase_23	AC	61 ms; 10,624 KB
testcase_24	AC	73 ms; 17,536 KB
testcase_25	AC	43 ms; 9,984 KB
testcase_26	AC	40 ms; 10,008 KB
testcase_27	AC	45 ms; 18,944 KB

ソースコード

#define _CRT_SECURE_NO_WARNINGS
#include<sstream>
#include<iostream>
#include<cstdio>
#include<cstdlib>
#include<cstring>
#include<climits>
#include<cmath>
#include<string>
#include<vector>
#include<set>
#include<map>
#include<queue>
#include<numeric>
#include<functional>
#include<algorithm>
#include<bitset>
#include<tuple>
#include<unordered_set>
#include<random>
using namespace std;
#define INF (1<<29)
#define rep(i,n) for(int i=0;i<(int)(n);i++)
#define all(v) v.begin(),v.end()
#define uniq(v) v.erase(unique(all(v)),v.end())



int n;
string s;
int c[100000], w[100000];
vector<int> edge[100000];
int sw;
long long ans;


void dfs1(int par,int u) {
	c[u] = w[u] = 0;
	if (s[u] == 'c')c[u]++;
	else w[u]++;
	for (int v:edge[u]) {
		if (v == par)continue;
		dfs1(u, v);
		c[u] += c[v];
		w[u] += w[v];
	}
}

void dfs2(int par, int u,int pc,int pw){
	int ew = sw;
	if (s[u] == 'w')ew--;
	for (int v : edge[u]) {
		if (v == par)continue;
		if (s[u] == 'w')ans += (long long)(c[v])*(ew - w[v]);
		dfs2(u, v, pc + c[u] - c[v], pw + w[u] - w[v]);
	}
	if (s[u] == 'w')ans += (long long)(pc)*(ew - pw);
}




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

	cin >> n >> s;
	rep(i,n-1) {
		int a, b;
		cin >> a >> b;
		a--; b--;
		edge[a].push_back(b);
		edge[b].push_back(a);
	}
	dfs1(-1, 0);
	sw = count(all(s), 'w');
	dfs2(-1, 0, 0, 0);
	cout << ans << endl;

	return 0;
}