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

题意:给你n个人,每个人都有一些兴趣,有相同兴趣的人会在一起,问最后会形成几个人群。


思路:并查集裸题,可以用一个vector来记录下每个兴趣有哪些人喜欢,这样读到某个人的某个兴趣时,可以直接

把这个人和这个兴趣存的人都关联起来。(最后记得要1到n都运行下Find,把每个人指向自己所在集合的祖先)



代码:

#include
#include
#include
#include
#include
using namespace std;
const int maxn = 1e3+5;
vector rec[maxn];
int n, pre[maxn], h[maxn], num[maxn];

int Find(int x)
{
    int r = x;
    while(pre[r] != r) r = pre[r];
    int i = x, j;
    while(pre[i] != r)
    {
        j = pre[i];
        pre[i] = r;
        i = j;
    }
    return r;
}

void join(int x, int y)
{
    int a = Find(x);
    int b = Find(y);
    if(a != b)
        pre[b] = a;
}

int main(void)
{
    while(cin >> n)
    {
        for(int i = 1; i <= n; i++)
            pre[i] = i;
        for(int i = 0; i < maxn; i++)
            rec[i].clear();
        for(int i = 1; i <= n; i++)
        {
            int x, tmp;
            scanf("%d:", &x);
            for(int j = 0; j < x; j++)
            {
                scanf("%d", &h[j]);
                for(int k = 0; k < rec[h[j]].size(); k++)
                    join(rec[h[j]][k], i);
            }
            for(int j = 0; j < x; j++)
                rec[h[j]].push_back(i);
        }
        int ans = 0;
        for(int i = 1; i <= n; i++)
        {
            Find(i);
            if(pre[i] == i)
                ans++;
        }
        printf("%d\n", ans);
        memset(num, 0, sizeof(num));
        for(int i = 1; i <= n; i++)
            num[pre[i]]++;
        sort(num+1, num+1+n);
        for(int i = n; i > n-ans; i--)
            printf("%d%c", num[i], i==n-ans+1 ? '\n' : ' ');
    }
    return 0;
}


你可能感兴趣的:(PAT,并查集,并查集)