Vision_数据结构_并查集

///定义：
/*
    并查集是一种树型的数据结构，用于处理一些不相交集合（Disjoint Sets）的合
并及查询问题。常常在使用中以森林来表示。
    集就是让每个元素构成一个单元素的集合，也就是按一定顺序将属于同
一组的元素所在的集合合并。
*/

///代码：

/*
**name:并查集
**function:合并查找分类（连通分量的个数）
*/
#include<bits/stdc++.h>
using namespace std;
int arr[1000];
/*
**非递归实现
**路径压缩优化
*/
int f(int x){
    int r,y;
    r = x;
    while(r!=arr[r]){
        r=arr[r];
    }
    y = x;
    int temp;
    while(y!=r){
        temp = arr[y];
        arr[y] = r;
        y  = temp;
    }
    return r;
}
/*
**递归实现
**路径压缩优化
*/
int F(int x){
    return x==arr[x]?x:arr[x] = F(arr[x]);
}
int main(){
    int n,x,y;
    while(cin>>n){
        for(int i = 0;i<=100;i++)arr[i] = i;
        for(int i = 0;i<n;i++){
            cin>>x>>y;
            if(f(x)!=f(y)){
                arr[f(x)] = f(y);
            }
        }
        for(int i = 1;i<=n;i++)cout<<" "<<f(i);
        cout<<endl;
    }
    return 0;
}

/*
**name:食物链（带权并查集）
**function:种类区分判断
*/
int N,K;
//T是信息的类型
int T[MAX],X[MAX],Y[MAX];
void solve(){
    ///初始化并查集
    ///元素x,x+N,x+2*n分别代表x->A,x->B,x->C
    init(N*3);
    int ans = 0;
    for(int i = 0;i<K;i++){
        int t = T[i];
        int x = X[i]-1,y=Y[i]-1;///把输入变成0~N-1的范围
        if(x<0||N<=x||y<0||N<=x){
            ans++;
            continue;
        }
        if(t==1){
            ///x和y属于同意一类
            if(same(x,y+N)||same(x,y+2*n)){
                ans++;
            }
            else {
                unite(x,y);
                unite(x+N,y+N);
                unite(x+2*N,y+2*N);
            }
        }
        else {
            ///x吃y
            if(same(x,y)||same(x,y+2*N)){///same(x,y+2*N)表示y吃x
                ans++;
            }
            else{
                unite(x,y+N);
                unite(x+N,y+2*N);
                unite(x+2*N,y);
            }
        }
    }
    return ans;
}

///扩展：有待学习更新带权并查集