1107. Social Clusters (30)--查并集

最新推荐文章于 2021-02-15 20:56:36 发布

原创最新推荐文章于 2021-02-15 20:56:36 发布 · 146 阅读

0 ·

CC 4.0 BY-SA版权

PAT 甲级同时被 2 个专栏收录

97 篇文章

订阅专栏

9‘’数据结构——树、堆、并查集

11 篇文章

订阅专栏

本文详细介绍了并查集算法的基本概念、实现原理及其在解决特定问题中的应用。通过具体的代码实例，展示了如何初始化并查集、查找元素所属集合以及合并两个集合的操作。此外，还介绍了一种优化方案——路径压缩，以提高算法效率。

#include <cstdio>
#include <cstdlib>
#include <iostream>
#include <algorithm>

#include <string>
#include <cstring>
#include <vector>
#include <queue>
#include <map>
#include <set>
#include <stack>

using namespace std;
int N;
int List[1005]={0};
int father[1005];
void init()
{
    for(int i=1;i<=N;i++)
        father[i]=i;
}

int findroot(int x)//查
{
    int p1=x;
    while(father[x]!=x)
        x=father[x];
    //x是根了

    //路径压缩
    while(father[p1]!=x)
    {
        int p2=p1;
        p1=father[p2];
        father[p2]=x;
    }

    return x;
}

void Union(int t1,int t2)//并
{
    int r1=findroot(t1);
    int r2=findroot(t2);
    if(r1!=r2)
        father[r1]=r2;
}

int book[1005]={0};
bool cmp(int a,int b)
{
    return a>b;
}
int main()
{
 //   freopen("in.txt","r",stdin);

    scanf("%d",&N);
    init();
    for(int i=1;i<=N;i++)
    {
        int k;
        scanf("%d:",&k);
        while(k--)
        {
            int t;
            scanf("%d",&t);
            if(List[t]==0)
                List[t]=i;
            else
                Union(List[t],i);
        }
    }

    for(int i=1;i<=N;i++)
    {
        int root=findroot(i);
        book[root]++;
    }

    vector<int> v;
    for(int i=1;i<=N;i++)
    {
        if(book[i]!=0)
            v.push_back(book[i]);

    }
    sort(v.begin(),v.end(),cmp);
    printf("%d\n",v.size());
    for(int i=0;i<v.size();i++)
    {
        printf("%d",v[i]);
        if(i!=v.size()-1)
            printf(" ");
    }


    return 0;
}