访问计划（C++）

题目描述

Farmer John 计划建造 N（1≤N≤10^5）个农场，用 N−1 条道路连接，构成一棵树（也就是说，所有农场之间都互相可以到达，并且没有环）。每个农场有一头奶牛，品种为更赛牛或荷斯坦牛之一。

Farmer John 的 M 个朋友（1≤M≤10^5）经常前来拜访他。在朋友 i 拜访之时，Farmer John 会与他的朋友沿着从农场 Ai 到农场 Bi 之间的唯一路径行走（可能有 Ai=Bi）。除此之外，他们还可以品尝他们经过的路径上任意一头奶牛的牛奶。由于 Farmer John 的朋友们大多数也是农场主，他们对牛奶有着极强的偏好。他的有些朋友只喝更赛牛的牛奶，其余的只喝荷斯坦牛的牛奶。任何 Farmer John 的朋友只有在他们访问时能喝到他们偏好的牛奶才会高兴。

请求出每个朋友在拜访过后是否会高兴。

输入格式

输入的第一行包含两个整数 N 和 M。 第二行包含一个长为 N 的字符串。如果第 i 个农场中的奶牛是更赛牛，则字符串中第 i 个字符为 'G'，如果第 i 个农场中的奶牛是荷斯坦牛则为 'H'。 以下 N−1 行，每行包含两个不同的整数 X 和 Y（1≤X,Y≤N），表示农场 X 与 Y 之间有一条道路。 以下 M 行，每行包含整数 Ai，Bi，以及一个字符 Ci。Ai 和 Bi 表示朋友 i 拜访时行走的路径的端点，Ci 是 'G' 或 'H' 之一，表示第 i 个朋友喜欢更赛牛的牛奶或是荷斯坦牛的牛奶。

输出格式

输出一个长为 M 的二进制字符串。如果第 i 个朋友会感到高兴，则字符串的第 i 个字符为 '1'，否则为 '0'。

样例输入

5 5
HHGHG
1 2
2 3
2 4
1 5
1 4 H
1 4 G
1 3 G
1 3 H
5 5 H

样例输出

10110

提示

在这里，从农场 1 到农场 4 的路径包括农场 1、2 和 4。所有这些农场里都是荷斯坦牛，所以第一个朋友会感到满意，而第二个朋友不会。 测试点 2-5 满足 N≤10^3,M≤2⋅10^3。

 思路

运用并查集，将相连的且有相同种类的农场并在一起。

判断时有三种情况：

1.起点和终点不在同一集合，说明路上一定会有两种奶牛，记为'1'；

2.起点和终点在同一集合，且奶牛种类为朋友要求的，记为'1'；

3.起点和终点在同一集合，但奶牛种类不是朋友要求的，记为'0'。

代码（可自行优化）

#include<bits/stdc++.h>
using namespace std;
int n, m, u, v,num=0;
int father[100100];
char q[100100],p[100100];



int find(int t) {
	if (father[t] != t) {
		father[t] = find(father[t]);
	}
	return father[t];
}

int main() {
	scanf("%d%d", &n, &m);
	for (int i = 1; i <= n; i++) {
		cin >> q[i];
	}
	for (int i = 1; i <= n; i++) {
		father[i] = i;
	}
	for (int i = 1; i < n; i++) {
		scanf("%d%d", &u, &v);
		if (q[u] == q[v]) {
			if (find(u) != find(v)) {
				father[find(u)] = find(v);
			}
		}
	}
	
	for (int i = 1; i <= m; i++) {
		int a, b;
		char c;
		scanf("%d%d", &a, &b);
		cin>>c;
		if (find(a) != find(b)) p[num]='1',num++;
		else {
			if (q[find(a)] == c) p[num]='1',num++;
			else p[num]='0',num++;
		}
	} 
    for(int i=0;i<num;i++) cout<<p[i];
    
	return 0;
}