POJ1182 食物链（并查集）

POJ1182食物链

#include<cstdio>
#include<iostream>
using namespace std;
const int MAX_K = 150010;
const int MAX_N = 1e5+20;
int N,K;
int T[MAX_K],X[MAX_K],Y[MAX_K];
int par[3*MAX_N],_rank[3*MAX_N];//T为信息类型
//初始化n个元素
void init(int n){
    for(int i=0;i<n;i++){
        par[i]=i;
        _rank[i]=0;
    }
}

//查询树的根
int find(int x){
    if(par[x]==x)
        return x;
    else
        return par[x]=find(par[x]);
}

//合并x和y所属的集合
void unite(int x,int y){
    x=find(x);
    y=find(y);
    if(x==y)
        return;
    if(_rank[x]<_rank[y]){
        par[x]=y;
    }else{
        par[y]=x;
    }
    if(_rank[x]==_rank[y])
            _rank[x]++;
}

//判断x和y是否属于同一类
bool same(int x,int y){
    return find(x)==find(y);
}

void solve(){
    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<=y){
            ans++;
            continue;
        }
                                                //第一种如果1 x y即：本条给定的信息是：x与y是同类
        if(t==1){
            //x和y属于同一类的信息
// 如果x吃y并且y也吃x，矛盾！说明此条信息有误
            if(same(x,y+N)||same(x,y+2*N)){
                ans++;
            }else{// 否则，x与y为同类的动物，则合并他们
                unite(x,y);
                unite(x+N,y+N);
                unite(x+2*N,y+2*N);
            }
        }else{                                      //第二种
            //x吃y的信息
 // 如果x与y是同类或者y吃x，又矛盾，说明此条信息有误
            if(same(x,y)||same(x,y+2*N)){
                ans++;
            }else{// 否则，x可以吃y，则将这条关系添加到集合中
                unite(x,y+N);
                unite(x+N,y+2*N);
                unite(x+2*N,y);
            }
        }
    }
    printf("%d\n",ans);
}


int main(){
    cin>>N>>K;
    for(int i=0;i<K;i++)
        scanf("%d%d%d",&T[i],&X[i],&Y[i]);
    //初始化并查集
    //元素x，x+N，x+2*N 分别代表x-A，x-B，x-C
    solve();
    return 0;
}

Sample Input

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

Sample Output

3