A1118 Birds in Forest (25分)

Posted 2020-12-06 tsruixi

一、技术总结

1. 题意是，给出N行，每行的第一个数字num，是接下来的鸟的数量。然后，如果两张图片中有相同的鸟的id号，代表这两张照片拍的是同一棵树。最后要求输出在树的数量和鸟的数量。同时给出k对数，查看这两只鸟是否在同一棵树上。
2. 抽象来理解就是考查并查集，需要将同一棵树上的鸟放进一个集合，关键是理解在合并的操作，我们只要将每行合并操作即可，如果其他行有出现的鸟，那么意味着这两行中的所有鸟会被合并到一个集合中。
3. 最后需要用一个bool类型的visit数组，存储出现过的鸟，用于统计鸟的数量，同时设置cnt，我们遍历所有visit中的鸟，如果出现该父节点是新出现的，那么说明出现一棵新树。

二、参考代码

#include<bits/stdc++.h>
using namespace std;
int father[10010];
bool visit[10010] = {false};
set<int> fa;//用于存放父亲结点 
int findF(int x){
	int a = x;
	while(x != father[x]){
		x = father[x];
	}
	while(a != father[a]){
		int z = a;
		a = father[a];
		father[z] = x;
	}
	return x;
}
void Union(int a, int b){
	int faA = findF(a);
	int faB = findF(b);
	if(faA != faB){
		father[faB] = faA;
	}
}
int main(){
	int n, k, cnt = 0, all_bird = 0;
	scanf("%d", &n);
	for(int i = 0; i < 10010; i++){
		father[i] = i;
	}
	for(int i = 0; i < n; i++){
		int num, id;
		scanf("%d %d", &num, &id);
		visit[id] = true;
		for(int j = 1; j < num;j++){
			int index;
			scanf("%d", &index);
			visit[index] = true;
			Union(id, index);
		}
	}
	for(int i = 0; i < 10010; i++){
		if(visit[i] && fa.find(findF(i)) == fa.end()){
			fa.insert(findF(i));
			cnt++;
		}
		if(visit[i]) all_bird++;
	}
	printf("%d %d
", cnt, all_bird);
	scanf("%d", &k);
	for(int i = 0; i < k; i++){
		int a, b;
		scanf("%d %d", &a, &b);
		if(findF(a) == findF(b))printf("Yes
");
		else printf("No
");
	}
	return 0;
}