poj1182(食物链)----带权并查集

这道题，刚开始开了3个并查集，各种关系把我搞得头晕，已提交就是wa，怎么也找不出错误，上网查题解发现是并查集姿势不对！！！

只需开一个并查集，只要能确定关系就扔进并查集里，其中每个元素与老大哥的关系无非三种：同类，被吃，吃人，分别用0,1,2来表示。

那么每来一次判断，先判断两个是否有关系，没关系就肯定是真话，join；

否则就得判断是否说谎；

d==1且ta与tb是同类，没说谎

d==2且ta与tb满足

三种可能

ta tb

0 1

1 2

2 0

合起来就是(rt[tb]-rt[ta]+3)%3==1)

那么没说谎

其余情况都是说谎。

下一个难点就是join函数

(d==1) rt[fx]=(rt[y]-rt[x]+3)%3;

(d==2) rt[fx]=(rt[y]-rt[x]+2)%3;

按照上面那种思路就可以得出，不懂的话可以看一些其他blog

http://blog.csdn.net/c0de4fun/article/details/7318642

http://wxdlut.blog.163.com/blog/static/128770158200982754311269/

#include <iostream>
#include <stdio.h>
#include <algorithm>
#include <stdlib.h>
#include <vector>
#include <string.h>
#include <queue>
#define ms(X) memset(X,0,sizeof(X))
typedef long long LL;
using namespace std;
int pre[50005],rt[50005];//0表示pre[x]和x同类，1表示x被吃，2表示x吃人
int fnd(int node)
{
if(pre[node]==node) return node;
int tmp=pre[node];
pre[node]=fnd(pre[node]);
rt[node]=(rt[tmp]+rt[node])%3;
return pre[node];
}
void join(int x,int y,int d)
{
int fx=fnd(x),fy=fnd(y);
if(fx!=fy) pre[fx]=fy;
if(d==1) rt[fx]=(rt[y]-rt[x]+3)%3;
else rt[fx]=(rt[y]-rt[x]+2)%3;
}
int main(int argc, char const *argv[])
{
int n,k,cnt=0;
cin>>n>>k;
int d,ta,tb;
ms(rt);
for(int i=1;i<=n;i++) pre[i]=i;
while(k--)
{
scanf("%d %d %d",&d,&ta,&tb);
if(ta>n||tb>n||(ta==tb&&d==2)) {cnt++;continue;}
if(fnd(ta)==fnd(tb)){
if(d==1&&rt[ta]==rt[tb]) continue;
else if(d==2&&(rt[tb]-rt[ta]+3)%3==1) continue;
else cnt++;
}
else join(ta,tb,d);
}
printf("%d",cnt );
return 0;
}