食物链POJ1182

void init(int n){
    for(int i=0;i<n;i++){fa[i]=i;rank[i]=0;//不对我这个初始化还跟题目有关的
    }
}
int find(int x){
    if(fa[x]==x) return x;
    else return fa[x]=find(fa[x]);
}
void unite(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y) return ;
    
    if(rank[x]<rank[y]) fa[x]=y;
    else{fa[y]=x;
        if(rank[x]==rank[y]) rank[x]++;
    }
    
}
bool same(int x,int y) return find(x)==find(y);你这样还不行的！

bool same(int x,int y)
{
return find(x)==find(y);}

#include<iostream>
using namespace std;
int n,k,T[1005],X[1005],Y[1005];
int fa[1005],rank[1005];
int ans=0;
void init(int n){
    for(int i=0;i<n;i++){fa[i]=i;rank[i]=0;
    }
}
int find(int x){
    if(fa[x]==x) return x;
    else return fa[x]=find(fa[x]);
}
void unite(int x,int y)
{
    x=find(x);
    y=find(y);
    if(x==y) return ;
    
    if(rank[x]<rank[y]) fa[x]=y;
    else{fa[y]=x;
        if(rank[x]==rank[y]) rank[x]++;
    }
    
}
bool same(int x,int y) 
{
return find(x)==find(y);
}
int main(){
    cin>>n>>k;
    for(int i=0;i<k;i++){cin>>T[i];cin>>X[i];cin>>Y[i];
    } 
    init(n*3);
    for(int i=0;i<k;i++){
        int t=T[i];
        int x=X[i]-1;
        int y=Y[i]-1;
        //变成了0到n-1的范围
        if(x<0||x>=n||y<0||y>=n){ans++;
            continue;} 
        if(t==1){
            if(same(x,y+n)||same(x,y+2*n)) ans++;
            else{
                unite(x,y);
                unite(x+n,y+n);
                unite(x+n*2,y+n*2); }
        }
        else{
            if(same(x,y)||same(x,y+2*n)) ans++;
            else{
                unite(x,y+n);
                unite(x+n,y+2*n);
                unite(x+2*n,y);
            }    
        }
    }
    cout<<ans<<endl;
}