problem
L3-003 社交集群 (30分) 当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式: 输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:
K i : h i [1] h i [2] … h i [K i ]
其中K i (>0)是兴趣爱好的个数,h i [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
给出n个人和他们的爱好列表,两个人有共同爱好即属于同一圈子输出有多少个圈子,以及每个圈子的人数
solution
第一感觉并查集模板,维护一个爱好集,最后统计有多少独立集合即可。统计圈子人数需要存在每个人的爱好,再跑一遍每次查询统计。用map统计排序挺麻烦的,查了一下资料按照key排序可以写比较函数,按照value排序只能转成vector<pair<int,int> >排。。
#include<bits/stdc++.h>
using namespace std
;
const int maxn
= 1010;
vector
<int>man
[maxn
];
bool cmp(pair
<int,int> a
, pair
<int,int> b
){return a
.second
>b
.second
;}
int fa
[maxn
+10];
void init(int n
){for(int i
= 0; i
<= n
; i
++)fa
[i
]=i
;}
int find(int x
){return x
==fa
[x
]?x
:fa
[x
]=find(fa
[x
]);}
void merge(int x
, int y
){x
=find(x
);y
=find(y
);if(x
!=y
)fa
[x
]=y
;}
int main(){
int n
; cin
>>n
;
init(maxn
);
for(int i
= 1; i
<= n
; i
++){
int k
; scanf("%d:",&k
);
int x
; cin
>>x
;
man
[i
].push_back(x
);
for(int j
= 2; j
<= k
; j
++){
int xx
; cin
>>xx
;
merge(x
,xx
);
man
[i
].push_back(xx
);
}
}
map
<int,int>ma
;
for(int i
= 1; i
<= n
; i
++){
ma
[find(man
[i
][0])]++;
}
vector
<pair
<int,int> >vec(ma
.begin(),ma
.end());
sort(vec
.begin(),vec
.end(),cmp
);
cout
<<vec
.size()<<endl
;
for(int i
= 0; i
< vec
.size()-1; i
++)
cout
<<vec
[i
].second
<<" ";
cout
<<vec
[vec
.size()-1].second
;
return 0;
}