并查集-poj2524

Posted 2020-08-25

以前只知道并查集的原理,并不知道具体实现有那些技巧.

一开始写并查集我把所有点的根全都初始化成0,然后find()等部分都写得很麻烦.

最普遍的写法是把每个点的根都设为自身,这样find就能写的很简便.具体看下面的代码实现

#include <cstdio>
#include <cstring>
int l[60000];
int fd(int a){
    return a==l[a]?a:(l[a]=fd(l[a]));
}
int main(){
    int n,m,a,b,ret,art,brt,t=0;
    while(scanf("%d %d",&n,&m) && n){
        t++;
        ret=0;
        for(int i=1;i<=n;i++){l[i]=i;}
        for(int i=0;i<m;i++){
            scanf("%d %d",&a,&b);
            art=fd(a),brt=fd(b);
            if(art!=brt){
                l[brt]=art;
            }
        }
        for(int i=1;i<=n;i++){
            if(l[i]==i)ret++;
        }
        printf("Case %d: %d\\n",t,ret);
    }
}

View Code