L3-003 社交集群并查集-蒲公英云

L3-003 社交集群并查集

女爷i 2022-03-10 21:09 260阅读 0赞

L3-003 社交集群（30 分）

当你在社交网络平台注册时，一般总是被要求填写你的个人兴趣爱好，以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。

输入格式：

输入在第一行给出一个正整数 N（≤1000），为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行，每行按以下格式给出一个人的兴趣爱好列表：

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

其中Ki(>0)是兴趣爱好的个数，hi[j]是第j个兴趣爱好的编号，为区间 [1, 1000] 内的整数。

输出格式：

首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔，行末不得有多余空格。

输入样例：

输出样例：

3
4 3 1
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<iostream>
#include<string>
#include<algorithm>
#include<map>
#include<set>
#include<queue>
#include<stack>
#include<vector>
using namespace std;
#define inf 0x3f3f3f3f
#define LL long long
int father[1005];
int find(int x){
    return father[x]==x?x:father[x]=find(father[x]);
}
void combine(int x,int y)
{
    father[find(x)]=find(y);
}
bool cmp(int x,int y)
{
    return x>y;
}
int main()
{
    int n,m,k,l,o,i,j,f;
    vector<int> v[1005];
    for(i=0;i<1005;i++)
    father[i]=i;    
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        scanf("%d:",&m);
        for(j=0;j<m;j++)
        {
            scanf("%d",&k);
            v[i].push_back(k);
        }
    }
    for(i=1;i<=n;i++)
    {
        for(j=i+1;j<=n;j++)
        {
            for(l=0;l<v[i].size();l++)
            {
                f=0;
                for(o=0;o<v[j].size();o++)
                {
                    if(v[i][l]==v[j][o])
                    {
                        f=1;
                        combine(j,i);
                        break;
                    }
                }
                if(f)break;
            }
        } 
    }
    int sum=0;
    map<int,int> ma;
    for(i=1;i<=n;i++)
    {
        if(father[i]==i)
        sum++;
        ma[find(father[i])]++;//一定要find哦！！！ 
        //printf("i:%d f[i]:%d\n",i,father[i]);
    }
    printf("%d\n",sum);
    map<int,int>:: iterator it;
    int s[1005],op=0;
    for(it=ma.begin();it!=ma.end();it++)
    {
        s[op++]=it->second;
    }
    sort(s,s+op,cmp);
    for(i=0;i<op-1;i++)
    printf("%d ",s[i]);
    printf("%d\n",s[op-1]);
}