POJ 1182 食物链（经典带权并查集 向量思维模式 很重要）

#include <iostream>
#include<algorithm>
#include <cstdio>
#include<cstring>
#include<math.h>
#include<memory>
using namespace std;
typedef long long LL;
#define max_v 50010
int sum;
struct node
{
    int pre;//前驱
    int relation;//和前驱的关系
    //p[i].relation表示i的根结点到i的偏移量
}p[max_v];
void make_set(int x)
{
    p[x].pre=x;//前驱默认自己
    p[x].relation=0;//关系默认0
}
int find_set(int x)
{
    int temp;
    if(x==p[x].pre)
        return x;
    temp=p[x].pre;
    p[x].pre=find_set(temp);
    p[x].relation=(p[x].relation+p[temp].relation)%3;
    return p[x].pre;
}
void union_set(int a,int b,int k)
{
    int root1=find_set(a);
    int root2=find_set(b);
    if(root1==root2)
    {
        if(k==1&&p[a].relation!=p[b].relation)
            sum++;
        if((k==2)&&((3-p[a].relation+p[b].relation)%3!=k-1))
            sum++;
        return ;
    }

    //合并
    p[root2].pre=root1;
    p[root2].relation=((k-1)+3+p[a].relation-p[b].relation)%3;
}
int main()
{
    int n,m,a,b,k;
    scanf("%d %d",&n,&m);

    sum=0;//假话数目

    for(int i=1;i<=n;i++)
         make_set(i);
    for(int i=0;i<m;i++)
    {
        scanf("%d %d %d",&k,&a,&b);
        if(a>n||b>n)//假话
        {
            sum++;
            continue;
        }
        if(k==2&&a==b)//假话
        {
            sum++;
            continue;
        }
        union_set(a,b,k);//判断当前话是否跟前面说过的话冲突
    }
    printf("%d\n",sum);
    return 0;
}