poj 1182--食物链（并查集，向量偏移）

并查集中特殊的向量偏移问题，

T^T一开始做过那个龙帮虎帮的题目（poj 1703），当时也就是简单的看了下，没有认真的学习这方面的问题，这个题就又不会了。

http://hi.baidu.com/tomspirit/item/d1f2a19b2aaf36d27a7f0158    这个博客里对于向量偏移的原理讲的很清楚。

下面给出具体实现的代码。

/*
    并查集，向量偏移
    rank[i]信息是i点跟当前并查集根节点的关系。
    在压缩合并的过程中同时维护rank[]数组的信息
*/
#include<iostream>
#include<stdio.h>
using namespace std;
const int Maxsize = 50100;

int father[Maxsize];

int rank[Maxsize];

void InitSet(int n)
{
    for(int i = 1; i <= n ; i++)
    {
        father[i] = i;
        rank[i] = 0;
    }
}

int FindSet(int a)
{
    if(father[a] != a)
    {
        int t = father[a];
        father[a] = FindSet(father[a]);
        rank[a] = (rank[a] +rank[t])%3;
    }
    return father[a];
}

void UnionSet(int a,int b,int d)
{
    int x = FindSet(a);
    int y = FindSet(b);
    father[x] = y;
    rank[x]=(rank[b]-rank[a]+3+d)%3;
}

int main()
{
    int N,K;
    scanf("%d %d",&N,&K);
    InitSet(N);
    int D,X,Y;
    int ans = 0;
    for(int i = 0 ; i < K ; i++)
    {
        scanf("%d %d %d",&D,&X,&Y);
        if(X > N || Y > N || (D == 2 && X == Y))
        {
            ans ++;
        }
        else
        {
            int x = FindSet(X);
            int y = FindSet(Y);
            if(x == y)
            {
                if((rank[X]-rank[Y]+3)%3 != D-1)
                {
                    ans++;
                }
            }
            else
            {
                UnionSet(X,Y,D-1);
            }
        }
    }
    printf("%d\n",ans);
    return 0;
}