7-7 社交集群 (30 分)

题目链接

并查集

#include <algorithm>
#include <bitset>
#include <cmath>
#include <cstdio>
#include <cstdlib>
#include <cstring>
#include <deque>
#include <functional>
#include <iostream>
#include <map>
#include <queue>
#include <set>
#include <stack>
#include <string>
#include <vector>

using namespace std;

int f[10010], a[10010];//f数组用来记录祖先节点，a数组用来记录每个人喜爱的集合中的某一个元素

int findn(int x)
{
    if (x == f[x])
        return x;
    else
        return f[x] = findn(f[x]);
}

bool book[10010];//标记兴趣爱好的编号是否出现过
int que[10010], k, ans[10010];//que数组用来最后排序每个集群中的人数，ans代表每个集群中的人数（有重复）

int main(int argc, char const *argv[])
{
    int n;
    cin >> n;
    for (int i = 1; i <= 1000; i++)//初始化
        f[i] = i;
    for (int i = 1; i <= n; i++)
    {
        int t, flag;
        scanf("%d:", &t);
        cin >> flag;//先读入一个兴趣编号
        a[i] = flag;//标记第i个人喜欢的兴趣编号代表
        book[flag] = 1;//标记这个兴趣编号出现过
        t--;//因为之前先读入了一个flag，所以t需要-1
        while (t--)
        {
            int x;
            cin >> x;
            book[x] = 1;//同理标记
            if (findn(x) != findn(flag))
                f[findn(x)] = findn(flag);
        }
    }
    int sum = 0;
    for (int i = 1; i <= 1000; i++)//便利所有兴趣编号
        if (book[i] == 1 && f[i] == i)//如果这个编号出现过同时这个编号是祖宗节点
            sum++;//集合数+1
    printf("%d\n", sum);//输出集合数
    for (int i = 1; i <= 1000; i++)
        if (book[i] == 1)
            f[i] = findn(i);//更新所有兴趣编号的祖宗节点
    for (int i = 1; i <= n; i++)//便利n个人的兴趣编号代表
        ans[f[a[i]]]++;//该集合人数+1
    for (int i = 1; i <= n; i++)
        if (ans[f[a[i]]] > 0)
        {
            que[k++] = ans[f[a[i]]];//添加到que数组中去
            ans[f[a[i]]] = 0;//清0，防止多次添加
        }
    sort(que, que + k);
    for (int i = k - 1; i > 0; i--)
        printf("%d ", que[i]);
    printf("%d\n", que[0]);
    return 0;
}