L3-003. 社交集群
在社交网络平台注册时,用户通常会输入自己的兴趣爱好,以便找到和自己兴趣相投的朋友。有部分兴趣相同的人们就形成了“社交集群”。现请你编写程序,找出所有的集群。
输入格式:
输入的第一行给出正整数N(<=1000),即社交网络中的用户总数(则用户从1到N编号)。随后N行,每行按下列格式列出每个人的兴趣爱好:
Ki: hi[1] hi[2] ... hi[Ki]
其中Ki(>0)是第i个人的兴趣的数量,hi[j]是第i个人的第j项兴趣的编号,编号范围为[1, 1000]内的整数。
输出格式:
首先在第一行输出整个网络中集群的数量,然后在第二行按非递增的顺序输出每个集群中用户的数量。数字间以1个空格分隔,行首尾不得有多余空格。
输入样例:
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
题解:
这道就是普通的并查集,不过中间使用了map进行优化。因为我需要从不同的并查组中得到更多关于组的性质。我就使用了map大大优化了查找的速度
#include <bits/stdc++.h>
using namespace std;
typedef long long ll;
int fa[1007];
int find1(int x){
return x == fa[x]?x:find1(fa[x]);
}
void unit(int x,int y){
int v = find1(x),u = find1(y);
if (v < u) fa[u] = v;
else fa[v] = u;
return ;
}
bool cmp(int x,int y){
return x > y;
}
int vis[1005];
int a[1007];
int main()
{
//ios::sync_with_stdio(false);
for (int i = 0;i <= 1005;i ++)fa[i] = i;
int n;
cin >> n;
for (int i = 1;i <= n;i ++){
int k;
scanf("%d:",&k);
cin >> a[i];
vis[a[i]] = 1;
for (int j = 1;j <= k-1;j ++){
int t;
cin >> t;
unit(t,a[i]);
vis[t] = 1;
}
}
map<int,int> se;
map<int,int>::iterator it,it2;
for (int i = 1;i <= n;i ++){
int x = find1(a[i]);
se[x] ++;
}
// it2 --;
cout << se.size() << endl;
vector<int> vp;
for (it = se.begin();it != se.end();it ++){
vp.push_back(it->second);
}
sort(vp.begin(),vp.end(),cmp);
for (int i = 0;i < vp.size();i ++){
cout << vp[i] << (i == vp.size()-1?'\n':' ');
}
}