0
点赞
收藏
分享

微信扫一扫

4-3-4 并查集 社交集群

代码小姐 2022-02-15 阅读 74
算法c++

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

输入格式:

输入在第一行给出一个正整数 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<bits/stdc++.h>
using namespace std;
int fa[1001];
int find(int x)
{
	if (fa[x] == x) return x;
	return fa[x] = find(fa[x]);
}
int main()
{
	int n;
//int cnt = 1;
	cin >> n;
	char a;
	int ff[1001];
	for (int i = 1; i <= 1001; i++)
		fa[i] = i;
	for (int i = 0; i < n; i++)
	{
		int k;
		cin >> k;
		getchar();
		cin >> ff[i];  //用一个数组为每个人做标记
//cnt = max(ff[i], cnt);
		for (int j = 1; j < k; j++)
		{
			int g;
			cin >> g;
//cnt = max(g, cnt);
			if (find(g) != find(ff[i])) fa[find(g)] = find(ff[i]);
		}
	}
	map <int, int> mp;
	vector<int>v;
	for (int i = 0; i < n; i++)
		mp[find(ff[i])]++;      //记录互不相交的集合及其元素数量(也就是每个同好群和它的人数)
//for (int i = 1; i <= cnt; i++)
      for (int i = 1; i <= 1000; i++)
		if (mp.count(i)) v.push_back(mp[i]);
	sort(v.begin(), v.end());
	cout << v.size() << endl;
	for (int i = v.size() - 1; i > 0; i--)
		cout << v[i] << " ";
	cout << v[0] << endl;
	return 0;
}

 

举报

相关推荐

0 条评论