poj 1182 食物链 并查集好题

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
struct node
{
    int root;
}nod[150010];
void initial(int n)
{
    for(int i=1;i<=3*n;i++){
        nod[i].root=i;
    }
}
int Find(int k)
{
    if(k!=nod[k].root){
        nod[k].root=Find(nod[k].root);
    }
    return nod[k].root;
}
void Union(int a,int b,int c,int d,int e,int f)
{
    nod[Find(b)].root=Find(a);
    nod[Find(d)].root=Find(c);
    nod[Find(f)].root=Find(e);
}
int main()
{
    int n,k;
    scanf("%d%d",&n,&k);
    initial(n);
    int ans=0;
    int cnt=1;
    for(int i=0;i<k;i++){
        int q,a,b;
        scanf("%d%d%d",&q,&a,&b);
        if(a>n || b>n ||a<=0 || b<=0){
            ans++;
            continue;
        }
        if(q==1){
           if(Find(a)==Find(b+n) || Find(a)==Find(b+2*n)){
              ans++;
           }
           else{
              Union(a,b,a+n,b+n,a+2*n,b+2*n);
           }
        }
        else{
            if(Find(a)==Find(b) || Find(a)==Find(b+2*n)){
                ans++;
            }
            else{
               Union(a,b+n,a+n,b+2*n,a+2*n,b);
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}