NOWCODER 白魔法师（图，并查集）

链接：https://ac.nowcoder.com/acm/problem/205460
来源：牛客网

思路：

图的并查集，只能染色一次，如果全是白点，输出大小n即可，否则只能将黑点染色成白点才有可能构成最大的白色连通块，维护每个白点连通块的大小，然后遍历所有黑点，将黑点直接相连的所有白色连通块的大小相加再加1（该黑点染成白点了），求出最大的ans。

AC代码：

#include<iostream>
#include<vector>
using namespace std;
const int maxn = 1e5+10;
int pre[maxn],n,cnt[maxn],x,y;
char str[maxn];
vector<int> v[maxn];
int xfind(int root) {
	return (root == pre[root] ? root : pre[root] = xfind(pre[root]));
}
int main() {
	cin>>n>>str;
	for(int i=0; i<=n; i++) pre[i] = i;
	for(int i=1; i<=n; i++) {
		if(str[i-1]=='W') cnt[i] = 1;
	}
	for(int i=0; i<n-1; i++) {
		cin>>x>>y;
		v[x].push_back(y),v[y].push_back(x);
		if(xfind(x) != xfind(y) && str[x-1]=='W' && str[y-1]=='W') {
			cnt[xfind(x)] += cnt[xfind(y)];
			pre[xfind(y)] = xfind(x);
		}
	}
	if(cnt[xfind(1)]==n) { //特判  全为白色的时候
		cout<<n<<endl;
		return 0;
	}
	int ans=0,sum;
	for(int i=1; i<=n; i++) {
		if(str[i-1]=='B') {
			sum = 1;
			for(int t:v[i]) sum += cnt[xfind(t)];
			ans = max(ans,sum);
		}
	}
	cout<<ans<<endl;
}