PAT-甲级-1118 Birds in Forest (25 分)

最新推荐文章于 2021-02-21 12:21:28 发布

原创最新推荐文章于 2021-02-21 12:21:28 发布 · 177 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

PAT 同时被 2 个专栏收录

63 篇文章

订阅专栏

甲级

60 篇文章

订阅专栏

本文深入探讨了一种基于并查集的鸟类群集算法实现，通过使用C++编程语言，详细解释了如何处理森林中鸟类之间的关系，包括如何构建并查集，查找根节点，以及合并不同鸟类群体的过程。文章提供了完整的代码示例，展示了算法的高效性和实用性。

1118 Birds in Forest (25 分)

#include <iostream>
#include <string>
#include <vector>
#include <set>
#include <map>
#include <queue>
#define rep(i,j,k)    for(int i=j;i<k;i++)
using namespace std;
const int maxn=1e4+10;
int fa[maxn];
int find(int x){
    int a = x;
    while(x != fa[x]){
        x = fa[x];
    }
    while(a != fa[a]){
        int z = a;
        a = fa[a];
        fa[z] = x;
    }
    return x;
}

void uni(int a,int b){
    int faa=find(a);
    int fab=find(b);
    if(faa < fab)
        fa[fab] = faa;
    else if(faa > fab)
        fa[faa] = fab;
}

int main(){
    int n,m,k,a,b;
    cin>>n;
    rep(i,0,10000){
        fa[i] = i;
    }
    set<int> st,ma;
    rep(i,0,n){
        cin>>k>>a;
        st.insert(a);
        rep(j,1,k){
            cin>>b;
            st.insert(b);
            uni(a,b);
        }
    }
    for(set<int>::iterator it=st.begin();it!=st.end();it++){
        ma.insert(fa[*it]);
    }
    printf("%d %d\n",ma.size(),st.size());
    cin>>m;
    rep(i,0,m){
        cin>>a>>b;
        printf("%s\n",find(a)==find(b)?"Yes":"No");
    }
    return 0;
}