Poj 食物链 解题报告 （种类并查集）

#include <cstdio>

const int MAXN = 50010;

int fake; //记录假话个数
int Set[MAXN]; //并查集的父节点
int relaWithFarther[MAXN]; //0 同类，1吃，2被吃，由这样的定义可知相反关系计算方法为，若A和B关系为X，则B和A的关系为（3-X）%3，注意这里的关系不是对称的。

void ini()
{
for(int i = 0; i < MAXN; ++i)
{
Set[i] = i;
relaWithFarther[i] = 0;
}
}

//一下“->”指吃
//注意如何计算本节点与树根（即将成为本节点的父节点）的关系，由向量相加原理可得
//relaWithFarther[node] = (relaWithFarther[node] + relaWithFarther[Set[node]])%3，
//这里模3是因为在这里如果A->B->C，即relationWithFarther[A]=2, relationWithFarther[B]=2，
//得到随后的relationWithFarther[A]=1。
int find(int node)
{
if(Set[node] == node)
return node;
int end = find(Set[node]);
int rela = relaWithFarther[node] + relaWithFarther[Set[node]];
relaWithFarther[node] = rela % 3;
Set[node] = end;
return end;
}

void Union(int nodeA, int nodeB, int relation)
{
int endA = find(nodeA);
int endB = find(nodeB);
//注意因为路径压缩，nodeA和nodeB的父节点已经转变。
if(endA == endB)
{
//此时应有的关系为rela，即有nodeA与endA的关系和nodeB与endB的关系可得nodeA和nodeB的关系，举个例子，A->B<-C可得A和B同类，有A与B的关系加上B和C的关系再模3得到。
int rela = relaWithFarther[nodeA] + (3-relaWithFarther[nodeB])%3;
rela = rela % 3;
if(relation != rela)
{
++fake;
return ;
}
}
Set[endA] = endB;
//见下图：
//A---B
//|   |
//C---D
//由向量相加可得relation(A,B) = relation(A,C)+relation(C,D)+relation(D,B)
relaWithFarther[endA] = ((3-relaWithFarther[nodeA])%3 + relaWithFarther[nodeB] + relation)%3;
}

int main()
{
ini();
fake = 0;
int m, n;
int nodeA, nodeB, relation;
scanf("%d %d", &m, &n);
for(int i = 0; i < n; ++i)
{
scanf("%d %d %d", &relation, &nodeA, &nodeB);
if((nodeA == nodeB && relation == 2) || nodeA > m || nodeB > m){
fake++;
continue;
}
Union(nodeA, nodeB, relation-1);
}
printf("%d\n", fake);
return 0;

}