L3-003 社交集群 (30 分)

最新推荐文章于 2022-03-17 00:33:58 发布

tcy今天长胖了吗

最新推荐文章于 2022-03-17 00:33:58 发布

阅读量121

点赞数

分类专栏： CCCC 数据结构文章标签：并查集

本文链接：https://blog.csdn.net/qq_49451105/article/details/116069377

版权

CCCC 同时被 2 个专栏收录

109 篇文章 3 订阅

订阅专栏

数据结构

13 篇文章 0 订阅

订阅专栏

这篇博客介绍了一个利用并查集解决社交集群问题的算法。作者首先初始化并查集，然后根据输入的爱好信息将用户连接。通过查找每个爱好所属的集合，将有共同爱好的用户连接在一起。最后，通过统计集合数量和成员数，输出社交集群的个数和各集群人数，实现了高效的人群分组。

摘要由CSDN通过智能技术生成

L3-003 社交集群 (30 分)

题目

题目链接

思路

开始想复杂了，以为用并查集可能做不了，后来发现其实就是并查集。至于部分爱好相同如何连接，我采用的方法是输入一个爱好之后去判断这个爱好的表当前是否为空，不为空就和表里的第一个人相连，并且无论是否是空的都放入表中。最后取出每个人的祖先放入set同时统计人数即可。

代码

#include <bits/stdc++.h>
#define INF 0x3f3f3f3f3f3f
#define pb push_back
using namespace std;
typedef pair<int, int> P;
const int N = 1e4 + 10;
const int M = 1e2 + 10;
typedef long long ll;

int par[N];
int rnk[N];

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

int find(int x)
{
    if(x == par[x])
        return x;
    return x = find(par[x]);
}

void unite(int x, int y)
{
    x = find(x);
    y = find(y);
    if(x == y)
        return ;
    if(rnk[x] > rnk[y])
    {
        par[y] = x;
    }
    else
    {
        par[x] = y;
        if(rnk[x] == rnk[y])
            rnk[y]++;
    }
}

vector<int> h[N];
vector<int> p[N];
set<int> st;
int num[N];
priority_queue<int> que;

int main()
{
    int n;
    scanf("%d", &n);
    init(1000);
    for(int i = 1; i <= n; i++)
    {
        int k;
        scanf("%d: ", &k);
        for(int j = 0, x; j < k; j++)
        {
            cin >> x;
            if(p[x].size())
            {
                unite(p[x][0], i);
            }
            p[x].pb(i);
        }
    }
    for(int i = 1; i <= n; i++)
    {
        st.insert(find(i));
        num[find(i)]++;
    }
    cout << st.size() << endl;
    bool flag = 0;
    for(auto i : st)
    {
        que.push(num[i]);
    }
    while(!que.empty())
    {
        cout << que.top();
        que.pop();
        if(que.size())
            cout << ' ';
    }
    return 0;
}