部落

最新推荐文章于 2024-04-17 16:26:18 发布

原创最新推荐文章于 2024-04-17 16:26:18 发布 · 184 阅读

0 ·

CC 4.0 BY-SA版权

并查集专栏收录该内容

5 篇文章

订阅专栏

本文介绍了一种通过并查集算法解决社区内部落统计及成员归属判断的问题。具体包括如何利用并查集进行数据结构的设计与实现，用以快速判断任意两人是否属于同一部落，并最终输出社区总人数和互不相交部落的数量。

时间限制

120 ms

内存限制

65536 kB

代码长度限制

8000 B

判题程序

Standard

作者

陈越

在一个社区里，每个人都有自己的小圈子，还可能同时属于很多不同的朋友圈。我们认为朋友的朋友都算在一个部落里，于是要请你统计一下，在一个给定社区中，到底有多少个互不相交的部落？并且检查任意两个人是否属于同一个部落。

输入格式：

输入在第一行给出一个正整数N（<= 10⁴），是已知小圈子的个数。随后N行，每行按下列格式给出一个小圈子里的人：

K P[1] P[2] ... P[K]

其中K是小圈子里的人数，P[i]（i=1, .., K）是小圈子里每个人的编号。这里所有人的编号从1开始连续编号，最大编号不会超过10⁴。

之后一行给出一个非负整数Q（<= 10⁴），是查询次数。随后Q行，每行给出一对被查询的人的编号。

输出格式：

首先在一行中输出这个社区的总人数、以及互不相交的部落的个数。随后对每一次查询，如果他们属于同一个部落，则在一行中输出“Y”，否则输出“N”。

输入样例：

输出样例：

10 2
Y
N

#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int f[10005];
int _find(int x)
{
    return x==f[x]?x:f[x]=_find(f[x]);
}
void _union(int x,int y)
{
    x=_find(x);
    y=_find(y);
    if(x!=y)
        f[x]=y;
}
int main()
{
    int n;
    scanf("%d",&n);
    int a[10005]={0},root[10005];
    int i,j,h=0,per=0,num=0;
    for(i=0;i<=10004;i++)
        f[i]=i;
    for(i=0;i<n;i++)
    {
        int k;
        scanf("%d",&k);
        int tt;
        scanf("%d",&tt);
        root[h++]=tt;
        if(a[tt]==0)
        {
            per++;
            a[tt]=1;
        }
        for(j=1;j<k;j++)
        {
            int t;
            scanf("%d",&t);
            if(a[t]==0)
            {
                per++;
                a[t]=1;
            }
            _union(tt,t);
        }
    }
    int mark[10005]={0};
    for(i=0;i<h;i++)
    {
        if(mark[_find(root[i])]==0)
        {
            num++;
            mark[_find(root[i])]=1;
        }
    }
    printf("%d %d\n",per,num);
    int q;
    scanf("%d",&q);
    for(i=0;i<q;i++)
    {
        int x,y;
        scanf("%d %d",&x,&y);
        if(_find(x)==_find(y))
            printf("Y\n");
        else
            printf("N\n");
    }
    return 0;
}