POJ1611（The Suspects）又见并查集

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

这貌似我做的第二道并查集的题，算是一道水题吧，题目的大概意思是找出与编号为0的人有

交集的人数。所以要将出现的集合中有交集的合并成大集合，最后输出0所在集合的人数。

 

贴代码：

 

#include<iostream>
using namespace std;
#define N 30005

int p[N],f[N];//f用来存集合元素的个数 

int find(int x)//并查集的根本，找父亲结点 
{
    return p[x] == x ? x : ( p[x] = find(p[x]) );
}

void merge(int a, int b)
{
    int x = find(a);
    int y = find(b);
    /*照顾0，所以把小的数作为父亲结点*/ 
    if(x < y) { 
        p[y] = x;
        f[x] += f[y];
    }
    else if(x > y) { //这里不能写else，因为x=y无需处理 
        p[x] = y;
        f[y] += f[x];
    }
}

int main()
{
    int n,m;
    int k,a,b;
    while(cin >> n >> m,n || m)
    {
        /*初始化并查集 */
        for(int i = 0; i < n; i++)
        {
            p[i] = i;
            f[i] = 1;
        }
        while( m--)
        {
            cin >> k;
            cin >> a;
            for(int j = 1; j < k; j++)
            {
                cin >> b;
                merge(a,b);//合并集合 
            }
        }
        cout << f[0] << endl;//输出0所在集合元素个数 
    }
    return 0;
}

   
可能是merge函数写的不够好，所以跑了63ms...

 现在又写了一遍：

/*Accepted    408K    0MS    C++    927B    2012-07-27 16:24:33*/
#include<cstdio>

const int MAXN = 300101;
int p[MAXN], cnt[MAXN];

int find_set( int x)
{
    return p[x] = p[x] == x ? x : find_set(p[x]);
}

void union_set( int x, int y)
{
    int nx = find_set(x), ny = find_set(y);
    if( nx < ny){ //以编号小的点为根
        p[ny] = nx;
        cnt[nx] += cnt[ny];
    }
    else if(nx > ny) {
        p[nx] = ny;
        cnt[ny] += cnt[nx];
    }
}

int main()
{
    int n, m, k;
    while( scanf( "%d%d", &n, &m), n || m)
    {
        for( int i = 0; i < n; i ++)
        {
            p[i] = i;
            cnt[i] = 1;
        }
        while( m --)
        {
            int x, y;
            scanf( "%d", &k);
            scanf( "%d", &x);
            for( int i = 1; i < k; i ++)
            {
                scanf( "%d", &y);
                union_set(x, y);
            }
        }
        printf( "%d\n", cnt[0]);
    }
    return 0;
}