poj1182->食物连->并查集

poj1182 食物链

思路来源，但是有所不同，详细异同点代码中有注释。

并查集小结：并查集算法主要是根据元素之间的关系去快速查询（Query )，我们在将元素加入（Merge ）我们的并查集中时要实时的更新关系函数（Relation 数组），弄清楚谁是谁的父节点（GetPar），谁是谁们的根节点（Parent 数组）。

#include<cstdio>

using namespace std;

const int Max_num = 50005;
int n,k;
int Parent[Max_num];
int Relation[Max_num];

void Init(){
    for(int i= 0;i<=n;i++){
        Parent[i] = i;
        //Relation[i] = 0 ->本身 1-> 吃掉父节点2->  被父节点吃掉
        Relation[i] = 0;
        }
}

int GetPar(int x){
    if(Parent[x] == x) return x;
    int pa = Parent[x];
    Parent[x] = GetPar(Parent[x]);
    //更新关系函数
    Relation[x] = (Relation[x]+Relation[pa])%3;
    return Parent[x];
}

bool Query(int x,int y,int d){
    int a = GetPar(x);
    int b = GetPar(y);
    if(a ==b){
        if(d == 1 && Relation[x] != Relation[y]) return false;
        if(d == 2){
            if(Relation[x] == 1 && Relation[y] != 0) return false;
            if(Relation[x] == 2 && Relation[y] != 1) return false;
            if(Relation[x] == 0 && Relation[y] != 2) return false;
        }
        return true;
    }
    /* 更新关系函数
    x->y 代表y是父节点
    0 ：没关系
    1 ：吃掉y
    2 ：被y吃掉

    用向量的进行巧解
    */
    //x吃掉y =》 b为a的父节点
    Parent[a] = b;
    if(d == 2){
    //有减法所以加三
        Relation[a] = (Relation[y]+1-Relation[x] +3)%3;
    }else{
        Relation[a] =(Relation[y]-Relation[x]+3)%3;
    }
    return true;
}


int main(){
    int x,y,d;
    scanf("%d%d",&n,&k);
    Init();
    int ans = 0;
    for(int i = 0;i<k;i++){
        scanf("%d%d%d",&d,&x,&y);
        if(x>n || y>n || (d==2 && x == y)){
            ans++;
            //continue忘了就是二倍
            continue;
        }
        if(!Query( x,y,d)){
            ans++;
        }
    }
    printf("%d\n",ans);
    return 0;
}