buct寒假集训——并查集

通畅工程

hdu1232

如果两个点集连通在一起，就合并在同一个集合，最终查看集合的总数。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int N=1010;
int fa[N];
int find(int x){
    if(x!=fa[x]) fa[x]=find(fa[x]);
    return fa[x]; 
}
void merge( int x,int y){
    if(find(x)!=find(y)){
        fa[find(x)]=find(y);
    }
}
int main(){
    int n,m;
    while(cin>>n){
        if(n==0) break;
        cin>>m;
        for( int i=1;i<=n;i++){
            fa[i]=i;
        }
        for( int i=1;i<=m;i++){
            int x,y;
            scanf("%d%d",&x,&y);
            merge(x,y);
        }
        int ans=0;
        for( int i=1;i<=n;i++){
            if(find(1)!=find(i)){
                merge(1,i);
                ans++;
            }
        }
        cout<<ans<<endl;
    }
}

方块栈

poj1988

维护size和到祖宗结点距离的并查集。
维护每个元素到栈底元素的距离。
维护的原理类似线段树的懒惰标记，每次只维护根节点的信息，子节点的信息仅当查找的时候才维护。

#include<bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int N=30010;
int fa[N],sum[N],under[N];
int find(int x){
    if(x!=fa[x]){
        find(fa[x]);
        under[x]+=under[fa[x]];
        fa[x]=find(fa[x]);
    } 
    return fa[x]; 
}
void merge( int x,int y){//将x移动到y的上方
    x=find(x),y=find(y);
    if(x!=y){
        fa[x]=y;
        under[x]+=sum[y];//这里加号可写可不写
        sum[y]+=sum[x];
    }
}
int main(){
    int n;
    cin>>n;
    for( int i=0;i<N;i++){
        fa[i]=i,sum[i]=1,under[i]=0;
    }
    for( int i=1;i<=n;i++){
        char s[2];
        int x,y;
        scanf("%s",s);
        if(s[0]=='M'){
            scanf("%d%d",&x,&y);
            merge(x,y);
        }
        else {
            scanf("%d",&x);
            find(x);
            printf("%d\n",under[x]);
        }
    }
    return 0;
}

食物链

poj1182
扩展域并查集/维护到根节点距离的并查集

维护到根节点距离的并查集原理：
当前节点到所处集合的 root 节点距离在模 3 意义下只会出现三种情况：0,1,2，我们将这三种情况分别指代为与 root 节点为 同类、吃根节点、被根节点吃。

这里使用扩展域并查集求解。
扩展域并查集代码：

#include<bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int N=50010;
int fa[N*3];
int find(int x){
    if(x!=fa[x]){
        fa[x]=find(fa[x]);
    } 
    return fa[x]; 
}
void merge( int x,int y){//将x移动到y的上方
    x=find(x),y=find(y);
    if(x!=y){
        fa[x]=y;
    }
}
int main(){
    int n,k;
    cin>>n>>k;
    for( int i=1;i<N*3;i++) fa[i]=i;
    int ans=0;
    for( int i=1;i<=k;i++){
        int d,x,y;
        scanf("%d%d%d",&d,&x,&y);
        if(x==y&&d==2) {
            ans++;
            continue;
        }
        if(x>n||y>n) {
            ans++;
            continue;
        }
        if(d==1){
            if(find(x)==find(y+n)||find(y)==find(x+n)){
                ans++;
                continue;
            }
            merge(x,y);
            merge(x+n,y+n);
            merge(x+n+n,y+n+n);
        }
        else {
            if(find(x)==find(y)||find(y)==find(x+n)){
                ans++;
                continue;
            }
            merge(x,y+n);
            merge(x+n,y+n+n);
            merge(x+n+n,y);
        }
 
    }
    cout<<ans<<endl;
}

帮派

poj1703
扩展域并查集

#include<bits/stdc++.h>
using namespace std;
typedef long long ll ;
const int N=400100;
int fa[N];
int find(int x){
    if(fa[x]!=x) {
        fa[x]=find(fa[x]);
    }
    return fa[x];
}
void merge( int a,int b){
    if(find(a)!=find(b)){
        fa[find(a)]=find(b);
    }
}
int main(){
    int  t;
    cin>>t;
    while(t--){
        int n,m;
        scanf("%d%d",&n,&m);
        for( int i=0;i<n+n+10;i++){
            fa[i]=i;
        }
        for( int i=1;i<=m;i++){
            char s[2];
            int a,b;
            scanf("%s%d%d",s,&a,&b);
            // cout<<a<<b<<find(a)<<find(b)<<endl;
            if(s[0]=='D'){
                merge(a,b+n);
                merge(b,a+n);
            }
            else {
                if(find(a)==find(b)||find(a+n)==find(b+n)){
                    printf("In the same gang.\n");
                }
                else if(find(a)==find(b+n)||find(b)==find(a+n)||n==2){
                    printf("In different gangs.\n");
                }
                else {
                    printf("Not sure yet.\n");
                }
            }
        }
    }
}