PAT程序设计考题——甲级1107（ Social clusters并查集） C++实现

点击打开pat链接

#include<iostream>

#include<math.h>

#include<algorithm>

#include<queue>

#include<map>

#include<stack>

#include<string>

#include<vector>

using namespace std;

#define INF 100000000

#define maxn 100010

int father[1010];

int isroot[1010]={0};

int coarse[1010]={0};

void init(int n)

{

 for(int i=0;i<=n;i++)

 {

  father[i]=i;

 }

}

int findfather(int x)

{ int a=x;

while(x!=father[x])

{

 x=father[x];

}

while(a!=father[a])

{int z=a;

 a=father[a];

 father[a]=x;

}

return x;

}

void Union(int x,int y)

{

 int m=findfather(x);

 int n=findfather(y);

 if(m!=n)  father[m]=n;

}

bool cmp(int a,int b)

{

 return a>b;

}

int main()

{ 

 int n,k,hobby;

 cin>>n;init(n);

 for(int i=1;i<=n;i++)

 {

  scanf("%d:",&k);

  for(int j=0;j<k;j++)

  {

   cin>>hobby;

 if(coarse[hobby]==0) coarse[hobby]=i;

 Union(i,father[coarse[hobby]]);

   }

   

 }//主体的功能已经结束

 for (int i=1;i<=n;i++)

 {

  isroot[findfather(father[i])]++;//同时统计人数

 }

    int num=0;

 for (int i=1;i<=n;i++)

 {

 if(isroot[i]!=0) num++;

 }

 cout<<num<<endl;

 sort(isroot+1,isroot+n+1,cmp);

 for(int i=1;i<=num;i++)

 {

  cout<<isroot[i];

  if(i<=num-1) cout<<" ";

 }

 return 0;

}