【poj1182】 食物链

http://poj.org/problem?id=1182 (题目链接)

题意

　　中文题

Solution

　　带权并查集。

　　神犇博客，秒懂

　　fa记录父亲，r记录与父亲的关系。%3运用的很巧妙。

代码

// poj1182
#include<algorithm>
#include<iostream>
#include<cstdlib>
#include<cstring>
#include<cstdio>
#include<cmath>
#define LL long long
#define inf 2147483640
#define Pi 3.1415926535898
#define free(a) freopen(a".in","r",stdin),freopen(a".out","w",stdout);
using namespace std;

int fa[1000010],r[1000010],n,k;

int find(int x) {
    if (x!=fa[x]) {
        int fx=find(fa[x]);
        r[x]=(r[x]+r[fa[x]])%3;
        fa[x]=fx;
    }
    return fa[x];
}
bool Union(int x,int y,int type) {
    int fx,fy;
    fx=find(x);fy=find(y);
    if (fx==fy) {
        if ((r[y]-r[x]+3)%3!=type) return 1;
        else return 0;
    }
    fa[fy]=fx;
    r[fy]=(r[x]-r[y]+type+3)%3;
    return 0;
}
int main() {
    scanf("%d%d",&n,&k);
    for (int i=1;i<=n;i++) fa[i]=i;
    memset(r,0,sizeof(r));
    int sum=0;
    while (k--) {
        int d,x,y;
        scanf("%d%d%d",&d,&x,&y);
        if (x>n || y>n || (x==y && d==2)) sum++;
        else if (Union(x,y,d-1)) sum++;
    }
    printf("%d",sum);
    return 0;
}