poj1182 食物链 带权并查集 偏移量

题目链接:

http://poj.org/problem?id=1182

题意:

题解:

大牛题解：http://blog.csdn.net/niushuai666/article/details/6981689
合并： d-1是偏移量，ra->rb, ra作为rb的父亲，注意方向【箭头是把右边合并到左边】
向量的思想：
if(ra!=rb)不在一个集合里，合并 fa[rb]=ra，还需要更新r[rb]，即与新的父节点(ra)的关系
r[rb] = ra->rb = ra->a+a->b+b->rb = r[a]+(d-1)-r[b];

fa[rb] = ra;
r[rb] = (r[a]+d-1-r[b]+3)%3;

if(ra==rb) 不需要合并了，判断是否矛盾，依然是向量的偏移量思想
a->b = a->ra+ra->b = -r[a]+r[b];
判断 -r[a]+r[b] 是否等于 d-1(偏移量)

if((r[b]-(d-1)+3)%3 != r[a]) ans++;

find操作，把信息传下来，路径压缩
向量思想： R表示根节点，t表示当前x的父节点，现在要实现R->x(把x压缩到根节点)
那么 R->x = R->t+t->x = r[t]+r[x]。 把这个值依次传下来。

r[x] = (r[x]+r[t])%3;

代码:

#include <iostream>
#include <cstdio>
#include <cstring>
#include <algorithm>
using namespace std;
typedef long long ll;
#define MS(a) memset(a,0,sizeof(a))
#define MP make_pair
#define PB push_back
const int INF = 0x3f3f3f3f;
const ll INFLL = 0x3f3f3f3f3f3f3f3fLL;
inline ll read(){
    ll x=0,f=1;char ch=getchar();
    while(ch<'0'||ch>'9'){if(ch=='-')f=-1;ch=getchar();}
    while(ch>='0'&&ch<='9'){x=x*10+ch-'0';ch=getchar();}
    return x*f;
}
//////////////////////////////////////////////////////////////////////////
const int maxn = 5e4+10;

int fa[maxn],r[maxn];

int find(int x){
    if(fa[x] == x) return x;
    int t = fa[x];
    fa[x] = find(fa[x]);
    r[x] = (r[x]+r[t])%3;
    return fa[x];
}

int main(){
    int n,k;
    scanf("%d%d",&n,&k);
    for(int i=0; i<=n; i++) fa[i]=i,r[i]=0;
    int ans = 0;
    for(int i=0; i<k; i++){
        int d,a,b,ra,rb;
        scanf("%d%d%d",&d,&a,&b);
        if(a>n || b>n) { ans++; continue;}
        if(d==2 && a==b) { ans++; continue;}
        ra = find(a),rb = find(b);
        if(ra == rb){
            if((r[b]-(d-1)+3)%3 != r[a]) ans++;
        }  
        else{
            fa[rb] = ra;
            r[rb] = (r[a]+d-1-r[b]+3)%3;
        }
    }

    cout << ans << endl;

    return 0;
}