巧妙的树上思维（并查集）

#include<bits/stdc++.h>
using namespace std;
const int N = 1e5+10;
int s[N];
char a[N];
void init(int n)
{
	for(int i=1;i<=n;i++)
	{
		s[i]=i;
	}
}
int find(int x)
{
	return s[x]!=x?s[x] = find(s[x]):s[x];
}
void merge(int x,int y)
{
	s[find(x)] = s[find(y)];
}
int main()
{
	cin.tie(0);
	cout.tie(0);
	ios::sync_with_stdio(false);
	int n,m;
	cin>>n>>m;
	init(n);
	for(int i=1;i<=n;i++)
	{
		cin>>a[i];
	}
	for(int i=1;i<n;i++)
	{
		int u,v;
		cin>>u>>v;
		if(a[u]==a[v])
		{
			merge(u,v);
		}
	}
	for(int i=1;i<=m;i++)
	{
		int u,v;
		char w;
		cin>>u>>v>>w;
		if(find(u)==find(v)&&w!=a[find(u)])
		{
			cout<<0;
		}else
		{
			cout<<1;
		}
	}
	return 0;
} 

P5836 [USACO19DEC]Milk Visits S - 洛谷 | 计算机科学教育新生态 (luogu.com.cn)