poj1182~食物链~带权并查集

#include<cstdio>
#include<iostream>
using namespace std;
#define maxsize 50050

int pre[maxsize];
int dist[maxsize];

int init(int n)
{
    for(int i=1;i<=n;i++)
    {
        pre[i]=i;
        dist[i]=0;
    }
}

int find(int x)
{
    if(x==pre[x]) return x;
    int tmp=find(pre[x]);
    dist[x]=(dist[pre[x]]+dist[x])%3;
    pre[x]=tmp;
    return tmp;
}


int Union(int w,int u,int v)
{
    int p=find(u);
    int q=find(v);
    if(p!=q)
    {
        pre[p]=q;
        dist[p]=(3+w+dist[v]-dist[u])%3;
    }
    else
    {
        if((dist[u]-dist[v]+3)%3!=w) return 0;
    }
    return 1;
}
int main()
{
    int n,k;
    scanf("%d %d",&n,&k);
        int cnt=0;
        init(n);
        int w,u,v;
        while(k--)
        {

            scanf("%d %d %d",&w,&u,&v);
            if(u>n||v>n||(w==2&&u==v)) {cnt++;}
            else if(!Union(w-1,u,v)) {cnt++;}
        }
        printf("%d\n",cnt);

    return 0;
}