当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 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;
const int N = 1e3 + 3;
int f[N], ch[N];
int Find(int x)
{
if(f[x] != x)
f[x] = Find(f[x]);
return f[x];
}
void Union(int x, int y)
{
int fx = Find(x);
int fy = Find(y);
if(fx != fy)
{
f[fx] = fy;
ch[fy] += ch[fx];
}
}
int main()
{
int n, k, h;
scanf("%d", &n);
vector<set<int> > vec(N);
for(int i = 1; i <= n; i++)
{
f[i] = i;
ch[i] = 1;
}
for(int i = 1; i <= n; i++)
{
scanf("%d:", &k);
while(k--)
{
scanf("%d", &h);
vec[i].insert(h);
for(int j = 1; j < i; j++)
{
if(vec[j].count(h) != 0)
{
Union(i, j);
break;
}
}
}
}
int cnt = 0, ans[N];
for(int i = 1; i <= n; i++)
{
if(Find(i) == i)
{
ans[cnt] = ch[i];
cnt++;
}
}
sort(ans, ans + cnt);
printf("%d\n", cnt);
for(int i = cnt-1; i >= 0; i--)
{
printf("%d", ans[i]);
if(i > 0)
printf(" ");
}
return 0;
}