当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 N(≤1000),为社交网络平台注册的所有用户的人数。于是这些人从 1 到 N 编号。随后 N 行,每行按以下格式给出一个人的兴趣爱好列表:
Ki: hi[1] hi[2] …hi[Ki]
其中Ki(>0)是兴趣爱好的个数hi[j]是第j个兴趣爱好的编号,为区间 [1, 1000] 内的整数。
输出格式:
首先在一行中输出不同的社交集群的个数。随后第二行按非增序输出每个集群中的人数。数字间以一个空格分隔,行末不得有多余空格。
输入样例:
1 2 3 4 5 6 7 8 9
| 8 3: 2 7 10 1: 4 2: 5 3 1: 4 1: 3 1: 4 4: 6 8 1 5 1: 4
|
输出样例:
思路
这道题是一个并查集的模板题,每次读入一个人兴趣爱好时,根据他的兴趣爱好和前一个有共同兴趣爱好的人进行合并,最后判断一共有多少个集合再输出即可。
代码
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48
| #include<iostream> #include<cstring> #include<vector> #include<algorithm> using namespace std; const int maxn = 1010; int p[maxn]; int number[maxn]; vector<int> v[maxn]; int st[maxn]; int n; int find(int x) { if(x != p[x]) p[x] = find(p[x]); return p[x]; } int main() { cin >> n; for(int i = 1; i <= n; i ++) p[i] = i; for(int i = 1; i <= n; i ++){ int num; scanf("%d:", &num); while(num --){ int x; cin >> x; if(!st[x]) st[x] = i; else{ int a = st[x], b = i; int fa = find(a), fb = find(b); if(fa != fb) p[fa] = fb; } } } int res = 0; for(int i = 1; i <= n; i ++){ number[find(i)] ++; if(find(i) == i) res ++; } cout << res << endl; sort(number + 1, number + n + 1); reverse(number + 1, number + n + 1); for(int i = 1; i <= res; i ++){ if(i != 1) cout << " "; cout << number[i]; } return 0; }
|