L3-003 社交集群并查集

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

L3-003 社交集群 （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<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]);
    }