当你在社交网络平台注册时,一般总是被要求填写你的个人兴趣爱好,以便找到具有相同兴趣爱好的潜在的朋友。一个“社交集群”是指部分兴趣爱好相同的人的集合。你需要找出所有的社交集群。
输入格式:
输入在第一行给出一个正整数 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
非并查集版本,20分代码,找不出哪里错了
#include <algorithm>
#include <iostream>
#include <vector>
using namespace std;
bool vis[1001] = {
false};
bool cmp(int a, int b) {
return a > b; }
int main() {
int n, k, id, cnt = 0;
char ch;
cin >> n;
int a[n + 1] = {
0};
vector<int> v[n + 1];
for (int i = 1; i <= n; i++) {
cin >> k >> ch;
bool flag = true;
vector<int> vv;
while (k--) {
cin >> id;
vv.push_back(id);
if (vis[id]) {
flag = false;
bool x = false;
for (int j = 1; j <= i - 1; j++) {
for (int p = 0; p < v[j].size(); p++) {
if (id == v[j][p]) {
a[j]++;
x = true;
break;
}
}
if (x)
break;
}
}
}
if (flag) {
for (int x = 0; x < vv.size(); x++) {
vis[vv[x]] = true;
v[i].push_back(vv[x]);
}
cnt++;
a[i]++;
}
}
cout << cnt << endl;
bool judge = true;
sort(a + 1, a + n + 1, cmp);
for (int i = 1; i <= n; i++) {
if (a[i]) {
if (judge) {
cout << a[i];
judge = false;
} else {
cout << " " << a[i];
}
} else {
break;
}
}
return 0;
}
并查集版本,30分,想到了并查集,但没有用一开始,用并查集写的也出了问题,因为没有想到怎么合并
#include <algorithm>
#include <iostream>
using namespace std;
int f[1001];
int flag[1001];
int num[2002];
int n;
bool cmp(int a, int b) {
return a > b; }
void init() {
for (int i = 1; i <= n; i++) {
f[i] = i;
}
}
int find(int x) {
if (x == f[x])
return x;
return f[x] = find(f[x]);
}
void merge(int x, int y) {
int a = find(x);
int b = find(y);
if (a != b)
f[b] = a;
}
int main() {
cin >> n;
init();
int k, id;
char ch;
for (int i = 1; i <= n; i++) {
cin >> k >> ch;
for (int j = 0; j < k; j++) {
cin >> id;
if (!flag[id])
flag[id] = i;
else
merge(i, flag[id]);
}
}
for (int i = 1; i <= n; i++) {
int x = find(i);
num[x]++;
}
int cnt = 0;
for (int i = 1; i <= n; i++) {
if (num[i])
cnt++;
}
cout << cnt << endl;
sort(num + 1, num + n + 1, cmp);
cout << num[1];
for (int i = 2; i <= n; i++) {
if (num[i])
cout << " " << num[i];
else
break;
}
return 0;
}