排位赛一 E Milk Visits

题目

Farmer John 计划建造 N 个农场，用 N-1 条道路连接，构成一棵树（也就是说，所有农场之间都互相可以到达，并且没有环）。每个农场有一头奶牛，品种为更赛牛或荷斯坦牛之一。
Farmer John 的 M 个朋友经常前来拜访他。在朋友 i 拜访之时，Farmer John 会与他的朋友沿着从农场 Ai到农场 Bi之间的唯一路径行走（可能有 Ai=Bi）。除此之外，他们还可以品尝他们经过的路径上任意一头奶牛的牛奶。由于 Farmer John 的朋友们大多数也是农场主，他们对牛奶有着极强的偏好。他的有些朋友只喝更赛牛的牛奶，其余的只喝荷斯坦牛的牛奶。任何 Farmer John 的朋友只有在他们访问时能喝到他们偏好的牛奶才会高兴。
请求出每个朋友在拜访过后是否会高兴。

解法

因为题目只有2种颜色，那么我们把同一种颜色连在一起的放在同一个连通块里面，也就是把两个点合并，查询的时候如果两个点没有公共祖先，那么他一定经过了2种颜色，如果是相同的公共祖先，那么我们只要判断它的节点的颜色是不是查询的颜色

代码

#include <iostream>
#include <cmath>
#include <queue>
#include <cstring>
#include <algorithm>
#include <cstdio>
using namespace std;

int n, m, from, to;
char ch;
char cow[100100];
int ans[100100];
int fa[100100];


int find(int x) {
	while(x != fa[x]) x = fa[x] = fa[fa[x]];
	return x;
}

void combine(int x, int y) {
	int a = find(x);
	int b = find(y);
	fa[a] = b;
}

int main() {
	cin >> n >> m;
	for(int i=1; i<=n; ++i) {
		fa[i] = i;
		cin >> cow[i];
	}
	
	for(int i=1; i<=n-1; ++i) {
		cin >> from >> to;
		if(cow[from]==cow[to]) combine(from, to);		
	} 
	int cnt = 0;
	for(int i=1; i<=m; ++i) {
		cin >> from >> to >> ch;
		if(find(from)==find(to) && cow[from]!= ch) ans[++cnt] = 0;
		else ans[++cnt] = 1;
	}
	
	for(int i=1; i<=cnt; i++) cout << ans[i];

	return 0;
}