1118 Birds in Forest (25 分)

并查集问题,考察的点有,集合的合并,集合个数的计算,各集合中元素数量统计。
坑点如下
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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值