POJ1611The Suspects并查集

http://poj.org/problem?id=1611

#include<iostream>
#include<math.h>
#include<stdio.h>
#include<string.h>
#include<math.h>
#include<algorithm>
using namespace std;
int pre[50000];
int n;
int m;
int find(int x)
{
    return x==pre[x]?x:pre[x]=find(pre[x]);
    //return
}
void unions(int x,int y)
{
    int fx=find(x);
    int fy=find(y);
    if(fx!=fy)
    {
        pre[fx]=fy;
    }
}//并查集方法;
void init()
{
    for(int i=0; i<=n; i++)
    {
        pre[i]=i;
    }
}
int main()
{
    while(cin>>n&&n)
    {
        memset(pre,0,sizeof(pre));
        init();
        cin>>m;
        int o,v;
        for(int i=0; i<m; i++)
        {
            cin>>o;
            int u;
            cin>>u;
            for(int j=1; j<o; j++)
            {
                cin>>v;
                unions(u,v);
            }
        }
        int ans=0;
        int ak=find(0);
        for(int i=0; i<n; i++)
        {
            if(find(pre[i])==ak)
            {
                ans++;
            }
        }
        cout<<ans<<endl;
    }

    return 0;
}