1107 Social Clusters

题目大意：

n个人，每个人都有k个兴趣，后面是兴趣种类的编号。有相同兴趣的两个人放到一个组里，两两组之间的人没有任何相同的兴趣，要求输出兴趣组数并按照非递增顺序输出兴趣小组的人数。

解题思路：

并查集的模板题，通过判断两个兴趣集合是否有交集判断这两个人是否属于同一组，开一个数组记录每棵树下的节点数量，在合并子节点时修改对应树的节点数量。
代码如下：

#include<iostream>
#include<cstdio>
#include<fstream>
#include<set>
#include<cmath>
#include<cstring>
#include<string>
#include<map>
#include<vector>
#include<iomanip>
#include<cstdlib>
#include<list>
#include<queue>
#include<stack>
#include<algorithm>
#define inf 0x3f3f3f3f
#define MOD 1000000007
#define mem0(a) memset(a,0,sizeof(a))
#define mem1(a) memset(a,-1,sizeof(a))
#define meminf(a) memset(a,inf,sizeof(a))
//vector ::iterator it;
//set<int>::iterator iter;
typedef long long ll;
typedef unsigned long long ull;
using namespace std;
int father[1100],n,son[1100];
struct student
{
  set<int> hobby;//兴趣集合
}s[1100];
void init()//初始化father数组
{
  for(int i=0;i<n;i++)
  {
    son[i]=1;
    father[i]=i;
  }
}
int getfather(int x)//找爹
{
  if(x==father[x])return x;
  else
  {
    father[x]=getfather(father[x]);//路径压缩，使每个子节点都指向树的总根节点
    return father[x];
  }
}
void unio(int a,int b)//合并两个节点
{
  int fa=getfather(a);
  int fb=getfather(b);
  if(fa!=fb)
  {
    father[fb]=fa;
    son[fa]+=son[fb];//节点数量
  }
}
int cmp(int a,int b)
{
  return a>b;
}
int main()
{
//  std::ios::sync_with_stdio(false);
//  cin.tie(0);
//  freopen("test.txt","r",stdin);
//  freopen("output.txt","w",stdout);
 scanf("%d",&n);
 init();
 for(int i=0;i<n;i++)
 {
   int num;
   scanf("%d: ",&num);
   for(int j=0;j<num;j++)
   {
     int tmp;
     scanf("%d",&tmp);s[i].hobby.insert(tmp);
   }
 }
 for(int i=0;i<n;i++)
 {
   for(int j=i+1;j<n;j++)
   {
     vector<int> v;//保存交集
     set_intersection(s[i].hobby.begin(),s[i].hobby.end(),s[j].hobby.begin(),s[j].hobby.end(),back_inserter(v));//获得交集
     if(v.size()) unio(i,j);//存在交集，合并
   }
 }
 int sum=0,data[1100];
 for(int i=0;i<n;i++)
 {
  if(i==father[i])//树的根节点
    data[sum++]=son[i];
 }
 printf("%d\n",sum);
 sort(data,data+sum,cmp);//从大到小排序
 printf("%d",data[0]);
 for(int i=1;i<sum;i++)printf(" %d",data[i]);
 printf("\n");
 return 0;
}