1. 程式人生 > >L3-1 社交叢集(30 分)

L3-1 社交叢集(30 分)

當你在社交網路平臺註冊時,一般總是被要求填寫你的個人興趣愛好,以便找到具有相同興趣愛好的潛在的朋友。一個“社交叢集”是指部分興趣愛好相同的人的集合。你需要找出所有的社交叢集。

輸入格式:

輸入在第一行給出一個正整數 N(1000),為社交網路平臺註冊的所有使用者的人數。於是這些人從 1 到 N 編號。隨後 N 行,每行按以下格式給出一個人的興趣愛好列表:

Ki: hi[1] hi[2] ... hi[Ki]

其中Ki(>0)是興趣愛好的個數,hi[j]是第j個興趣愛好的編號,為區間 [1, 1000] 內的整數。

輸出格式:

首先在一行中輸出不同的社交叢集的個數。隨後第二行按非增序輸出每個叢集中的人數。數字間以一個空格分隔,行末不得有多餘空格。

輸入樣例:

8
3: 2 7 10
1: 4
2: 5 3
1: 4
1: 3
1: 4
4: 6 8 1 5
1: 4

輸出樣例:

3
4 3 1

很明顯是並查集,比賽的時候我在想怎麼判斷兩個人有相同的興趣,,結果還是暴力。。先暴力搜尋有相同興趣的兩個人,剩下的就是並查集了

#include <bits/stdc++.h>

using namespace std;
const int N = 1e3 + 3;

int f[N], ch[N];

int Find(int x)
{
    if(f[x] != x)
        f[x] = Find(f[x]);
    return f[x];
}

void Union(int x, int y)
{
    int fx = Find(x);
    int fy = Find(y);
    if(fx != fy)
    {
        f[fx] = fy;
        ch[fy] += ch[fx];
    }
}

int main()
{
    int n, k, h;
    scanf("%d", &n);
    vector<set<int> > vec(N);
    for(int i = 1; i <= n; i++)
    {
        f[i] = i;
        ch[i] = 1;
    }
    for(int i = 1; i <= n; i++)
    {
        scanf("%d:", &k);
        while(k--)
        {
            scanf("%d", &h);
            vec[i].insert(h);
            for(int j = 1; j < i; j++)
            {
                if(vec[j].count(h) != 0)
                {
                    Union(i, j);
                    break;
                }
            }
        }
    }
    int cnt = 0, ans[N];
    for(int i = 1; i <= n; i++)
    {
        if(Find(i) == i)
        {
            ans[cnt] = ch[i];
            cnt++;
        }
    }
    sort(ans, ans + cnt);
    printf("%d\n", cnt);
    for(int i = cnt-1; i >= 0; i--)
    {
        printf("%d", ans[i]);
        if(i > 0)
            printf(" ");
    }
    return 0;
}