Poj 1611 The Suspects

题意

n个人，给出m组（总人数小于等于n），求0所在组的人数。

要点

这是一道并查集的简单题，要熟悉查找操作的递归写法、压缩路径写法、非递归压缩路径写法以及合并的优化和非优化写法。在获取某个节点所在的集一定要通过查找，而不是直接通过数组S[i]，因为在合并操作时，可能只改变了根节点的集，而没有更新当前集所有的节点，s[i]可能不是最终对应的集。

代码

#include <stdio.h>
const int maxn=30010;
int s[maxn];
int height[maxn];
void init_set(){
    for(int i=0;i<maxn;++i){
        s[i]=i;
        height[i]=0;
    }
       
}

int find_set(int x){
#if 0/*递归写法*/
    if(x!=s[x]) s[x]=find_set(s[x]);
    return s[x];
#else/*非递归写法*/
    int r=x;
    while(r!=s[r]) r=s[r];
    int i=x,j;
    while(i!=r){
        j=s[i];
        s[i]=r;
        i=j;
    }
    return r;
#endif
}

void union_set(int x,int y){
    x=find_set(x);
    y=find_set(y);
    if(x==y) return;
    if(height[x]==height[y]){
        height[x]+=1;
        s[y]=x;
    }else{
        if(height[x]<height[y]){
           s[x]=y;
        }
        else {
            s[y]=x;
        }
    }
}

int main(){
    int n,m,k,x,y,res;
    while(~scanf("%d%d",&n,&m)&&n!=0&&!(n==0&&m==0)){
        init_set();
        for(int i=0;i<m;i++){
            scanf("%d",&k);
            if(k<1) continue;
            scanf("%d",&x);
            for(int j=1;j<k;j++){
                scanf("%d",&y);
                union_set(x,y);
            }
        }
        res=0;
        /*错误做法
         for(int i=0;i<n;i++)
            if(s[i]==s[0]) res++
         错误原因，因为在union_set时，若两个点的根结点高度不同，只改变了根结点的集（s[i]对应的值），并不是这个集中的所有节点的s[i]改变了
         */
        int s0=find_set(0);
        for(int i=0;i<n;i++){
            if(find_set(i)==s0) res++;
        }
        printf("%d\n",res);
    }
    return 0;
}