CCCC题集 L3-003. 社交集群（并查集

最新推荐文章于 2019-12-17 01:11:01 发布

原创最新推荐文章于 2019-12-17 01:11:01 发布 · 264 阅读

0 ·

CC 4.0 BY-SA版权

online judge Others 同时被 2 个专栏收录

68 篇文章

订阅专栏

数据结构并查集

12 篇文章

订阅专栏

本文介绍了一种使用并查集算法寻找社交网络中具有共同兴趣爱好的用户集群的方法。通过输入用户的兴趣爱好，程序能够识别出不同的社交集群，并统计各集群的规模。

题目描述

在社交网络平台注册时，用户通常会输入自己的兴趣爱好，以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序，找出所有的集群。

输入

输入的第一行给出正整数N（<=1000），即社交网络中的用户总数（则用户从1到N编号）。随后N行，每行按下列格式列出每个人的兴趣爱好：

Ki: hi[1] hi[2] … hi[Ki]

其中Ki（>0）是第i个人的兴趣的数量，hi[j]是第i个人的第j项兴趣的编号，编号范围为[1, 1000]内的整数。

输出

首先在第一行输出整个网络中集群的数量，然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔，行首尾不得有多余空格。

样例

输入样例：
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

题意

简单的并查集可以用set去下重鉴于set不能使用下标访问这里用了vector

AC代码

#include <bits/stdc++.h>
using namespace std;

#define LL long long
#define CLR(a,b) memset(a,(b),sizeof(a))

const int MAXN = 1e4+10;
int arr[MAXN];
int par[MAXN];
vector<int> v;

int find(int x) {
    if(x == par[x])
        return x;
    return par[x] = find(par[x]);
}
void unite(int x, int y) {
    x = find(x);
    y = find(y);
    if(x != y)
        par[x] = y;
}

int main() {

    for(int i = 0; i < MAXN; i++)
        par[i] = i;
    int n;
    cin >> n;
    for(int i = 1; i <= n; i++) {
        int x, y, z;
        cin >> x;
        getchar();
        cin >> y;
        v.push_back(y);
        for(int j = 1; j < x; j++) {
            cin >> z;
            unite(y, z);
        }
    }
    for(int i = 0; i < v.size(); i++)
        arr[find(v[i])]++;
    v.clear();
    for(int i = 0; i < MAXN; i++) {
        if(arr[i]) v.push_back(arr[i]);
    }
    sort(v.begin(), v.end(), greater<int>());
    int len = v.size();
    cout << len << endl;
    for(int i = 0; i < len-1; i++) {
        cout << v[i] << ' ';
    }
    cout << v[len-1] << endl;
    return 0;
}