1118 Birds in Forest (25 分)

田园菜菜狗

于 2021-03-10 14:10:04 发布

阅读量81

点赞数

分类专栏： PAT (Advanced Level)

本文链接：https://blog.csdn.net/qq_43616564/article/details/114634132

版权

PAT (Advanced Level) 专栏收录该内容

4 篇文章

订阅专栏

本文通过一个具体的程序实例展示了并查集算法的应用场景与实现细节。该程序利用并查集来处理一系列元素间的连接关系，并解决如何判断任意两个元素是否属于同一集合的问题。此外，还涉及了输入解析、数据结构初始化等实用编程技巧。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include<bits/stdc++.h>
using namespace std;
vector<int> father,countn;
int findFather(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 fa=findFather(a);
	int fb=findFather(b);
	if(fa!=fb){
		father[fa]=fb;
	}
}
int main(){
	int n,k,q,a,b,c;
	map<int,int> mp;
	scanf("%d",&n);
	int course[10001]={0};
	father.resize(n+1);
	countn.resize(n+1);
	for(int i=1;i<=n;i++){
		father[i]=i;
	}
	for(int i=1;i<=n;i++){
		scanf("%d",&k);
		while(k--){
			scanf("%d",&a);
			mp[a]++;
			if(course[a]==0){
				course[a]=i;
			}	
			Union(i,findFather(course[a]));		
		}
	}
	for(int i=1;i<=n;i++){
		countn[findFather(i)]++;
	}	
	int cnt=0;
	for(int i=1;i<=n;i++){
		if(countn[i]!=0){
			cnt++;
		}
	}
	printf("%d %d\n",cnt,mp.size());
	scanf("%d",&q);
	for(int i=0;i<q;i++){
		scanf("%d %d",&b,&c);
		if(findFather(course[b])==findFather(course[c])){
			printf("Yes\n");
		}else{
			printf("No\n");
		}
	}
	return 0;
}