L2-024 部落 (25 分)

题目链接

题目大意

 给你一个n表示小圈子的个数，然后是每个小圈子人的编号，最后是q次查询，以及被查询的一对人的编号。要求输出社区的总人数和互不相交的部落的个数，以及每次查询的两人是否属于同一个部落进行判断。

解题思路

 1.初始化，将属于同一部落的进行合并

 2.用set计算共有多少人

 3.有多少个互不相交的部落，遍历所有人如果那个人的祖先就是自己，多一个部落

 4.任意两个人如果祖先相同就是同一部落

坑点

 无

代码

#include<bits/stdc++.h>
using namespace std;
const int N=1e4+10;
int fa[N];//根节点 
int find(int i)//查找
{
    if(i==fa[i])//找根节点 
    {
        return i;//找到的话返回 i 
    }
    else{//若不属于这个节点 
        fa[i]=find(fa[i]);//查找该点的根节点 
        return fa[i];//返回根节点 
    }
}
void unio(int i,int j)//合并
{
    int x1=find(i);//将 i 赋值给 x1 
    int x2=find(j);//将 j 赋值给 x2 
    fa[x2]=x1;//具有相同的合并 
}
int main()
{
    int n;
    cin>>n;//n个部落 
    for(int i=0;i<=N;i++)//初始化
    {
        fa[i]=i;//将根节点赋值 
    }
    set<int>st;//set计算共有多少人
    for(int i=0;i<n;i++)
    {
        int k,x,x1;
        cin>>k;//各个部落的人的数量 
        if(k>0)//有人时 
        {
            cin>>x;//先将第一个人存入部落方便后面的合并
            st.insert(x);//将人存入st中，表示是根节点 
        }
        for(int j=1;j<k;j++)//接下来 k-1个 
        {
            cin>>x1;//读入编号 
            st.insert(x1);//再次插入进去 
            unio(x,x1);//依次合并为同一祖先
        }
    }
    int ans=0;//计算共有多少不同部落
    for(int i=1;i<=st.size();i++)//遍历 
    {
        if(fa[i]==i)//如果那个人的祖先就是自己，多一个部落
        {
            ans++;//计数 
        }
    }
    cout<<st.size()<<" "<<ans<<endl;//输出部落的数量和不相交的人的数量 
    int f;
    cin>>f;//f 次查询 
    while(f--)//循环 
    {
        int a,b;
        cin>>a>>b;//读入一对的编号 
        if(find(a)==find(b))//如果两个人的祖先相同
        {
            cout<<"Y"<<endl;
        }
        else{
            cout<<"N"<<endl;//祖先不同输出 N 
        }
    }
    return 0;
}

总结

并查集的应用