当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:
输出格式:
首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔,行末不得有多余空格。
输入样例:
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 <iostream>
#include <vector>
#include <set>
#include <algorithm>
using namespace std;
const int MAX = 1e3 + 10;
int anc[MAX];
int n;
vector<vector<int>> like;
int findd(int x)
{
if (anc[x] == x)
return x;
return anc[x] = findd(anc[x]);
}
void unionn(int a, int b)
{
int a_anc = findd(a);
int b_anc = findd(b);
if (a_anc != b_anc)
anc[a_anc] = b_anc;
return;
}
bool isSimilar(int a, int b) //判断两人爱好是否有相同点
{
for (int i : like[a])
for (int j : like[b])
if (i == j)
return true;
return false;
}
bool cmp(int a, int b)
{
return a > b;
}
int main()
{
cin >> n;
for (int i = 1; i <= n; i++)
anc[i] = i;
like.resize(n + 1);
for (int i = 1; i <= n; i++) //记录每个人的爱好
{
vector<int> tem;
int k; cin >> k; getchar();
while (k--)
{
int x; cin >> x;
tem.push_back(x);
}
like[i] = tem;
}
for (int i = 1; i <= n; i++) //爱好相似的两人合并
for (int j = i + 1; j <= n; j++)
if (isSimilar(i, j))
unionn(i, j);
set<int> res; //社交集群,统计个数
int cot[MAX] = { 0 }; //社交集群人数
for (int i = 1; i <= n; i++)
{
int x = findd(i);
res.insert(x);
cot[x]++;
}
cout << res.size() << endl;
sort(cot + 1, cot + n + 1, cmp); //有效编号1~n
for (int i = 1; i <= res.size(); i++)
{
cout << cot[i];
if (i != res.size()) cout << ' ';
}
return 0;
}
注意事项:
并查集,题目给了3000ms,实际最大用时9ms哈哈哈~
如有问题,欢迎提出。