并查集之寻找某元素所在集合的元素个数----the suspects

另外建立一个数组NUM[],每次FIND(x)如果相等之后都将NUM[X]++，简单题。。。

#include<cstdio>
#include<iostream>
using namespace std;

#define N 40000
int pre[N];
int num[N];
int rank[N];

void init()
{
    for(int i=0; i<N; i++)
    {
        num[i]=1;
        pre[i]=i;
        rank[i]=0;
    }
}

int find(int x)
{
    int r=x;
    while(r!=pre[r])
        r=pre[r];
    int i=x,j;
    while(i!=r)
    {
        j=pre[i];
        pre[i]=r;
        i=j;
    }
    return r;
}
void merge(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fy==fx)   return;
    if(rank[fx]>rank[fy])
    {
        pre[fy]=fx;
        num[fx]+=num[fy];
    }
    else
    {
        pre[fx]=fy;
        if(rank[fx]==rank[fy])
            rank[fy]++;
        num[fy]+=num[fx];
    }
}
int main()
{

    int n,m;
    while(~scanf("%d%d",&n,&m))
    {
        init();
        if(n==0&&m==0)  break;
        if(m==0)
        {
            printf("1
");
            continue;
        }
        while(m>0)
        {
            m--;
            int t,first,next;
            cin>>t>>first;
            for(int i=1; i<t; i++)
            {
                scanf("%d",&next);
                merge(first,next);
                first=next;
            }
        }
        int x=find(0);
        cout<<num[x]<<endl;
    }
    return 0;
}