PAT 1107 Social Clusters

最新推荐文章于 2021-01-29 22:20:56 发布

原创最新推荐文章于 2021-01-29 22:20:56 发布 · 285 阅读

0 ·

CC 4.0 BY-SA版权

PAT 专栏收录该内容

63 篇文章

订阅专栏

博主表示遇到一道简单题目，在编写并查集的一个函数时出现失误。写demo运行未发现问题，由此反思以后要更加严谨。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

挺简单的题目吧，但是并查集的一个函数写挫了。写demo跑了一下也没啥问题啊，以后还是老实一点吧。

#include <cstdio>
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;

const int maxn = 1010;
int father[maxn];
int isroot[maxn]={0};
int course[maxn]={0};
int n;

int findf(int x)
{
    int a = x;
    while(x!=father[x])
    {
        x=father[x];
    }
    while(a!=father[a])//path compression
    {
        int z=a;
        a=father[a];
        father[z]=x;

    }
    return x;
}

void onion(int x, int y)//问题所在
{
    int xx=findf(x);
    int yy=findf(y);
    if(xx!=yy)
        father[x]=y;
//    x=findf(x);
//    y=findf(y);
//    if(x!=y)
//        father[x]=y;
}

void init()
{
    for(int i=1;i<=n;i++)
    {
        father[i]=i;
        isroot[i]=0;
    }
}

bool cmp(int a,int b)
{
    return a>b;
}

int main()
{
    int i,j,k,x,y;
    init();
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        scanf("%d:",&k);
        for(j=0;j<k;j++)
        {
            scanf("%d",&x);
            if(course[x]==0)
                course[x]=i;
            onion(i,findf(course[x]));
        }

    }
    for(i=1;i<=n;i++)
    {
        isroot[findf(i)]+=1;
    }
    int ans = 0;
    for(i=1;i<=n;i++)
    {
        if(isroot[i]!=0)
            ans++;
    }
    printf("%d\n",ans);

    sort(isroot+1,isroot+n+1,cmp);
    for(i=1;i<ans;i++)
        printf("%d ",isroot[i]);
    printf("%d",isroot[ans]);

    return 0;
}