1118 Birds in Forest(25 分)

版权声明:https://github.com/godspeedcurry 欢迎加好友哦 https://blog.csdn.net/qq_38677814/article/details/82082258

并查集入门,关键是想到并查集,我还想了其他操作好久发现没有什么想法
一个合并一个查找,就该想到并查集嘛
对于每一棵树上的鸟,可以将第一只作为已知在树上的鸟,把后面的鸟捉过来即可

#include <bits/stdc++.h>
using namespace std;
int fa[102000];
int find(int x){
    return fa[x]==x?x:fa[x]=find(fa[x]);
}
void Union(int a,int b){
    int fx=find(a);
    int fy=find(b);
    if(fx!=fy) fa[fy]=fx;
}
set<int>s;
int main(){
    int n;
    cin>>n;
    for(int i=1;i<=101000;i++) fa[i]=i;
    for(int i=1;i<=n;i++){
        int k;cin>>k;
        int boss;
        cin>>boss;
        s.insert(boss);
        for(int j=1;j<k;j++){
            int another_bird;
            cin>>another_bird;
            s.insert(another_bird); 
            Union(boss,another_bird);
        }
    }
    int ans=0;
    for(int i=1;i<=s.size();i++){
        if(find(i)==i) ans++;
    }
    printf("%d %d\n",ans,s.size());
    int q;
    cin>>q;
    while(q--){
        int x,y;
        cin>>x>>y;
        if(find(x)!=find(y)){
            printf("No\n");
        }
        else{
            printf("Yes\n");
        }
    }
    return 0;
}

猜你喜欢

转载自blog.csdn.net/qq_38677814/article/details/82082258