PAT（甲级）1107 Social Clusters (30point(s))

题目

题目链接

思路

题目大意：根据兴趣爱好将人群分组，但有一点比较绕，就是A和B有同样的兴趣，那么他们在一个组里；B和C也有相同的兴趣，他们在一个组里，但是即使A和C没有相同的兴趣，他们也在同一个组里；
肯定要用并查集做了，并查集主要有两个函数找父亲和合并两个集合，除此之外，还要搞清在什么时候才会进行合并操作，题意是两个人有相同兴趣，则合并这两个人所在的集合。明显的，当每次读入一个人的兴趣时，如果这个兴趣已经有人喜欢过了，那么就要进行合并操作了；
要统计每个集合的人数，一般开一个数组isroot[i] 用来保存以第i个节点为根节点的集合元素个数，进行合并操作的时候更新就可以了；

代码

#include <iostream>
#include <vector>
#include <stack>
#include <queue>
#include <algorithm>
#include <cstring>
#include <string>
#include <math.h>
using namespace std;
const int maxn = 1010;
//father用于存储父节点，isRoot存储以这个节点为根时子元素的个数，
//hobby存储一个喜欢人的id
int father[maxn], isRoot[maxn], hobby[maxn];
int n;

int findFather(int child){
     int a = child;
     while(father[child] != child){
          child = father[child];
     }
     //压缩路径，再遍历一遍，直接将老大填入
     int z = 0;
     while(father[a] != a){
          z = a;
          a = father[a];//保存上一父节点
          father[z] = child;
     }
     return child;
}
//合并任意两个人，先分别找到这两个人的老大，看他们的老大是不是同一个人
void Union(int a, int b){
     int faA = findFather(a);
     int faB = findFather(b);
     //如果老大不是同一个人，那么直接修改两个老大间的关系
     //这道题的特殊性，对于一个刚进来的节点来说，它的老大肯定是他自己
     if(faA != faB){
          //修改老大的同时修改每个集合的人数
          father[faA] = faB;
          isRoot[faB] += isRoot[faA];
          isRoot[faA] = 0;
     }
}

bool cmp(int a, int b){
     return a > b;
}
int main()
{
     scanf("%d", &n);
     //初始化father数组
     for(int i = 1; i <= n; i ++) father[i] = i;
     fill(isRoot, isRoot + maxn, 1);
     fill(hobby, hobby + maxn, 0);
     int k, t;
     for(int i = 1; i <= n; i ++){
          scanf("%d:", &k);
          for(int j = 0; j < k; j ++){
               scanf("%d", &t);
               //如果这个兴趣目前还没人喜欢，直接修改hobby数组
               if(hobby[t] == 0) hobby[t] = i;
               //如果有人喜欢，那么就要合并
               else Union(i, hobby[t]);
          }
     }
     //找到不为零的个数
     int ans = 0;
     for(int i = 1; i <= n; i ++){
          if(isRoot[i] != 0) ans ++;
     }
     printf("%d\n", ans);
     //排序，输出
     sort(isRoot + 1, isRoot + n + 1, cmp);
     for(int i = 1; i <= ans; i ++){
          printf("%d", isRoot[i]);
          if(i != ans) printf(" ");
     }
     system("pause");
     return 0;
}