并查集问题,考察的点有,集合的合并,集合个数的计算,各集合中元素数量统计。
坑点如下
1,在函数内定义数组时注意要赋值0。
整体代码如下
#include<bits/stdc++.h>
using namespace std;
int father[10010];
int vis[10010]={false};
int cnt[10010];
int n,q;
int find(int x){
while(father[x]!=x){
x=father[x];
}
return x;
}
void Union(int a,int b){
int fac,fab;
fac=find(a);
fab=find(b);
if(fac>fab)father[fac]=fab;
else father[fab]=fac;
}
int main(){
scanf("%d",&n);
int k;
for(int i=0;i<10010;i++)father[i]=i;
for(int i=0;i<n;i++){
int id,temp;
scanf("%d%d",&k,&id);
vis[id]=true;
for(int i=0;i<k-1;i++){
scanf("%d",&temp);
vis[temp]=true;
Union(id,temp);}
}
int numtree=0,numbird=0;
for(int i=0;i<10010;i++){
if(vis[i]==true)
cnt[find(i)]++;
}
for(int i=0;i<10010;i++){
if(cnt[i]!=0){
numtree++;
numbird+=cnt[i];
}
}
int q;
printf("%d %d\n",numtree,numbird);
scanf("%d",&q);
int v1,v2;
for(int i=0;i<q;i++){
scanf("%d%d",&v1,&v2);
if(find(v1)==find(v2)){printf("Yes\n");}
else{printf("No\n");}
}
return 0;
}
另一种解法
#include<bits/stdc++.h>
using namespace std;
int father[10010],isroot[10010];
set<int>s;
int findfather(int x){
while(father[x]!=x){
x=father[x];
}
return x;
}
int Union(int a,int b){
int fa=findfather(a);
int fb=findfather(b);
return fa>fb?father[fa]=fb:father[fb]=fa;
}
int main(){
int n,k,f,p,q,v1,v2,cnt=0;
cin>>n;
for(int i=0;i<10010;i++)father[i]=i;
for(int i=1;i<=n;i++){
scanf("%d",&k);
scanf("%d",&f);
s.insert(f);
k--;
while(k--){
scanf("%d",&p);
s.insert(p);
Union(f,p);
}
}
for(int i=0;i<10010;i++)
{
if(s.find(i)!=s.end())
isroot[findfather(i)]++;
}
for(int i=0;i<10010;i++){
if(isroot[i]!=0)cnt++;
}
cin>>q;
printf("%d %d\n",cnt,int(s.size()));
while(q--){
scanf("%d%d",&v1,&v2);
if(findfather(v1)==findfather(v2))printf("Yes\n");//此处看的是根结点是否相同,而不是父亲结点。
else printf("No\n");
}
return 0;
}