Submission #3923773

---

Source Code Expand

#include<iostream>
#include<string>
#include<cstdio>
#include<vector>
#include<cmath>
#include<algorithm>
#include<functional>
#include<iomanip>
#include<queue>
#include<ciso646>
#include<random>
#include<map>
#include<set>
#include<complex>
#include<bitset>
#include<stack>
#include<unordered_map>
using namespace std;
typedef long long ll;
typedef unsigned int ui;
const ll mod = 1000000007;
const ll INF = (ll)1000000007 * 1000000007;
typedef pair<int, int> P;
#define stop char nyaa;cin>>nyaa;
#define rep(i,n) for(int i=0;i<n;i++)
#define per(i,n) for(int i=n-1;i>=0;i--)
#define Rep(i,sta,n) for(int i=sta;i<n;i++)
#define rep1(i,n) for(int i=1;i<=n;i++)
#define per1(i,n) for(int i=n;i>=1;i--)
#define Rep1(i,sta,n) for(int i=sta;i<=n;i++)
typedef long double ld;
typedef complex<ld> Point;
const ld eps = 1e-3;
const ld pi = acos(-1.0);
typedef pair<ll, ll> LP;
typedef pair<ld, ld> LDP;

int n;
vector<int> G[1 << 17];
bool exi[1 << 17];

bool used[1 << 17];
int root;
string c;
bool maketree(int id) {
	used[id] = true;
	bool res = false;
	rep(i, G[id].size()) {
		int to = G[id][i];
		if (used[to])continue;
		if (maketree(to))res = true;
	}
	if (c[id] == 'W')res = true;
	if (!res)exi[id] = false;
	return res;
}
int deg[1 << 17];
int ans = 0;
void calcdeg(int id) {
	if (!exi[id])return;
	rep(j, G[id].size()) {
		int to = G[id][j];
		if (!exi[to])continue;
		deg[id]++;
	}
	ans += 2 + ((c[id] == 'W') ^ (deg[id] % 2));
}
void all_calcdeg() {
	rep(i, n)calcdeg(i);
}
int d[1 << 17];
int mostfar(int s) {
	fill(used, used + n, false);
	queue<int> q; used[s] = true; q.push(s);
	int res = s;
	d[s] = (c[s]=='W')^(deg[s]%2);
	while (!q.empty()) {
		int x = q.front(); q.pop();
		rep(i,G[x].size()) {
			int to = G[x][i];
			if (used[to])continue;
			if (!exi[to])continue;
			used[to] = true; q.push(to);
			d[to] = d[x] + ((c[to] == 'W')^(deg[to]%2));
		}
	}
	int ma = 0;
	rep(j, n) {
		if (ma < d[j]) {
			ma = d[j]; res = j;
		}
	}
	return res;
}

int main() {
	int num = 0;
	cin >> n; fill(exi, exi + n, true);
	rep(i, n - 1) {
		int a, b; cin >> a >> b; a--; b--;
		G[a].push_back(b);
		G[b].push_back(a);
	}
	cin >> c;
	root = -1;
	rep(i, n) {
		if (c[i] == 'W') {
			root = i; num++;
		}
	}
	if (root < 0) {
		cout << 0 << endl; return 0;
	}
	else if (num == 1) {
		cout << 1 << endl; return 0;
	}
	maketree(root);
	all_calcdeg(); ans -= 2;
	int l = mostfar(root), r = mostfar(l); l = mostfar(r);
	//cout << l << " " << r << endl;
	//rep(i, n)cout << d[i] << endl;
	cout << ans-2*d[l] << endl;
	//stop
	return 0;
}

Submission Info

Judge Result