POJ 1611 The Suspects 并查集问题

题目链接：http://acm.hust.edu.cn/vjudge/contest/view.action?cid=66964#problem/B

题意：SARS问题，把与0号属于同一个集合的 Suspect 计算出来。

题意很简单，不过又犯了老错误，把两个点是否属于同一个集合，用pre[i]==pre[j]来判断了，导致WA，看来得牢记：并查集的两个点是否属于同一个集合用的是find，不是pre！！！还有一点，这里增加了一个vis数组，用来标志改点是否在之前已经属于其他集合，如果是的话，则把其他的加入该集合，否则，把该点加入其他集合。

代码：

#include<iostream>
#include<cstdio>
#include<cstring>

using namespace std;

int n,m;
int pre[30010],vis[30010];

int find(int x){
    int t=x;
    while(pre[t]!=t) t=pre[t];
    while(x!=t) pre[x]=t,x=pre[x];
    return t;
}

int main(){
    //freopen("D:\\in.txt","r",stdin);
    while(cin>>n){
        cin>>m;
        if(!n && !m) break;
        for(int i=0;i<=n;i++) pre[i]=i;
        memset(vis,0,sizeof(vis));//点是否已经在之前已经出现过
        for(int i=0;i<m;i++){
            int k,t,s;scanf("%d",&k);
            for(int j=0;j<k;j++){
                scanf("%d",&t);
                if(j==0) {s=t;vis[s]=1;continue;}
                if(vis[t]) pre[find(s)]=find(t);
                else pre[find(t)]=find(s);
                s=t;
                vis[s]=1;
            }
        }
        int cnt=1;
        for(int i=1;i<n;i++) if(find(i)==find(0)) cnt++;//不是pre数组！！
        cout<<cnt<<endl;
    }
    return 0;
}