poj 1182 食物链

分析

　　并查集,需要维护集合内元素与根节点的关系(吃?被吃?同类)

　　有点类似向量的想法

　　如果x y属于同一集合

　　

　　

　　如果x y不属于同一集合

 

　　dis[x]=0表示x与根节点同类,1表示根节点吃x,2表示x吃根节点.

　　

#include<cstdio>
#include<cstring>
#define N 50010
using namespace std;
int root[N],dis[N],n,m;
void init(){
    for(int i=1;i<=n;i++){
        root[i]=i;
        dis[i]=0;
    }
}
int find(int x){
    if(x==root[x])return x;
    int rx=root[x];
    root[x]=find(root[x]);
    dis[x]=(dis[x]+dis[rx])%3;
    return root[x];
}
int main(){
    scanf("%d%d",&n,&m);
        init();
        int ans=0;
        while(m--){
            int d,x,y;
            scanf("%d%d%d",&d,&x,&y);
            if(x>n||y>n){
                ++ans;
                continue;
            }
            if(d==1){
                if(find(x)!=find(y)){
                    dis[root[x]]=(dis[y]-dis[x]+3)%3;//注意这两句的更新顺序...
                    root[root[x]]=root[y];
                }
                else{
                    if((dis[y]-dis[x]+3)%3)
                        ++ans;
                }
            }
            else{
                if(find(x)!=find(y)){
                    dis[root[x]]=(dis[y]-dis[x]+4)%3;
                    root[root[x]]=root[y];
                }
                else{
                    if((dis[y]+1-dis[x]+3)%3)
                        ++ans;
                }
            }
        }
        printf("%d\n",ans);
    
    return 0;
}