当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:
其中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
题解
并查集的题。设置两个数组,一个用来存为同一个社交群体的用户,一个存爱好。
将爱好数组 hobby 全部设置为 -1
然后每次遇到一个新的爱好,先看这个爱好对应的数值值是不是 -1
- 是 -1:说明这个爱好是第一次出现,之前没有人有过相同的爱好,把 hobby 数组中对应的值改为这个人的编号
- 不是 -1:说明有人有相同的爱好,将这两个人的集合合并
最后排个序就可以了
#include <bits/stdc++.h>
using namespace std;
ostream &sp(ostream &output);
bool cmp(int a, int b)
{
return a > b;
}
int father[1010];
int hobby[1010];
int findFather(int x)
{
if (x == father[x])
return x;
return findFather(father[x]);
}
int main()
{
std::ios::sync_with_stdio(false);
// fill(hobby, hobby + 1010, -1);
for (int i = 0; i < 1010; ++i)
{
father[i] = i;
hobby[i] = -1;
}
int n;
cin >> n;
char colon;
for (int i = 0; i < n; ++i)
{
int habitsNum;
cin >> habitsNum;
cin >> colon;
for (int j = 0; j < habitsNum; ++j)
{
int hobbyID;
cin >> hobbyID;
if (hobby[hobbyID] == -1) //说明还没有人和这个人有一样的爱好
hobby[hobbyID] = i;
else
{
// 把有相同爱好的人合并到一个并查集中
int a = findFather(i);
int b = findFather(hobby[hobbyID]);
if(a != b)
{
father[a] = b;
}
}
}
}
int cnt = 0;
int ans[1010];
fill(ans, ans + 1010, 0);
for (int i = 0; i < n; ++i)
{
// cout << "findFather(i) = " << findFather(i) << endl;
if (i == findFather(i))
cnt++;
ans[findFather(i)]++;
}
sort(ans, ans + 1010,cmp);// sort(ans,ans+cnt,cmp) 只能过 2 个点
cout << cnt << endl;
for (int i = 0; i < cnt - 1; ++i)
cout << ans[i] << “ ”;
cout << ans[cnt - 1] << endl;
return 0;
}