L3-003 社交集群 (30分)

 

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

 

 

分析:并查集

 

#include <iostream>
#include <vector>
#include <algorithm>
using namespace std;
vector<int> father, isRoot;
int cmp1(int a, int b){
         return a > b;
}
int findFather(int x) {
    return x == father[x] ? x : father[x] = findFather(father[x]); 
}
void Union(int a, int b) {
 int faA = findFather(a);
 int faB = findFather(b);
 if( faA != faB)
     father[faA] = faB;
}
int main() {
 int n, k, t, cnt = 0;
 int course[1009] = {0};
 scanf("%d", &n);
 father.resize(n + 1);
 isRoot.resize(n + 1);
 for( int i = 1; i <= n; i++)
      father[i] = i;
 for( int i = 1; i <= n; i++) {
      scanf("%d:", &k);
      for( int j = 0; j < k; j++) {
           scanf("%d", &t);
           if(   course[t] == 0)
                 course[t] = i;
           else  Union( i, findFather(course[t]) );
      }
  }
  for( int i = 1; i <= n; i++)
       isRoot[findFather(i)]++;
  for( int i = 1; i <= n; i++ ) {
       if( isRoot[i] != 0)
           cnt++;
  }
  printf("%d\n", cnt);
  sort(isRoot.begin(), isRoot.end(), cmp1);
  for( int i = 0; i < cnt; i++) {
       printf("%d", isRoot[i]);
       if( i != cnt - 1) 
	      printf(" ");
  }
 return 0; 
}

 

已标记关键词 清除标记
相关推荐
©️2020 CSDN 皮肤主题: 程序猿惹谁了 设计师:白松林 返回首页