团体程序设计天梯赛 L3-003 社交集群

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

题解 

并查集，枚举每一种爱好，如果这种爱好有超过1个人，那么就把这些人合并，最后合并完再遍历一遍，就能找到不同的集群，用优先队列来进行存储，非增输出。

#include<bits/stdc++.h>
using namespace std;
#define ll long long
#define endl '\n'
int n;
//vector<int> g[1005];
map<int,vector<int> > mp;
map<int,int> ans;
int f[1005];
priority_queue<int,vector<int> ,less<int> > q;
int find(int x)//并查集模板
{
if(x==f[x])
{
return x;
}
f[x]=find(f[x]);
return f[x];
}
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
f[i]=i;
int k;
//cin>>k;
scanf("%d:",&k);
for(int j=1;j<=k;j++)
{
int x;
cin>>x;
//g[i].push_back_back(x);
mp[x].push_back(i);//放到每个爱好里
}
}
for(auto k:mp)//遍历爱好
{
if(k.second.size()>1)
{
vector<int> temp=k.second;
int u=find(temp[0]);
for(int i=1;i<temp.size();i++)
{
int v=find(temp[i]);
                //合并
if(u!=v)
{
f[v]=u;
}
}
//int u=find(k.second[0]);

}
}
for(int i=1;i<=n;i++)
{
ans[find(i)]++;
}
for(auto k:ans)
{
q.push(k.second);
}
int len=q.size();
int op=0;
cout<<len<<endl;
while(!q.empty())//优先队列非递增输出
{
op++;
if(op<len)
{
cout<<q.top()<<" ";
}
else
{
cout<<q.top()<<endl;
}
q.pop();
}
return 0;
}

原文地址：https://blog.csdn.net/X_StarX/article/details/137832901

免责声明：本站文章内容转载自网络资源，如本站内容侵犯了原著者的合法权益，可联系本站删除。更多内容请关注自学内容网（zxcms.com）！