poj 1182（食物链）并查集

第一种说法是"1 X Y"，表示X和Y是同类。 

第二种说法是"2 X Y"，表示X吃Y。 

对于第一种说法，需要检查是否前面出现过2 x y，或 2 y x。对于第二种说法，需要检查是否出现过1 x y,2 y x。

如果对于每个出现的说法都去遍历检查，时间复杂度会过高。

由于需要判断说法是同类，自然的想到并查集，但对于这儿来说，不仅要组织是否是同一类，还要组织捕食关系。

如何组织呢，并查集里的情况都是满足不矛盾的情况。比如说，1 x y是否为真，如果为真则加入并查集，我们需

要判断 2 x y,2 y x和1 x y是否也在这个并查集树上，如果在，则1 x y为假，否则加入。每个i有三种可能的物种。

x,x+N,x+2*N，表示A,B,C。

x和y属于同一类，如果成立，合并X Y,X+N Y+N,X+2*N Y+2*N。

X吃Y,如果成立，合并X Y+N,X+N Y+2N,X+2N Y。

/*
poj 1182
*/
#include <stdio.h>
// using namespace std;
#define N 50010
int n,k;
int par[3*N];
int rank[3*N];

void init()
{
for(int i=0;i<3*N;i++)
{
rank[i]=0;
par[i]=i;
}
}

int find(int x)
{
if(par[x]==x)
return x;
return par[x]=find(par[x]);
}

void unite(int x,int y)
{
x=find(x);
y=find(y);
if(x==y)
return;
if(rank[x]<rank[y])
par[x]=y;
else
{
par[y]=x;
if(rank[x]==rank[y]) rank[y]++;
}
}

bool same(int x,int y)
{
return find(x)==find(y);
}

int main(int argc, char const *argv[])
{
// freopen("input","r",stdin);
init();
scanf("%d%d",&n,&k);
int x,p,q;
int ans=0;
for(int i=0;i<k;i++)
{
scanf("%d%d%d",&x,&p,&q);
if(p>0 && p<=n && q>0 && q<=n)
{
if(x==1)
{
if(same(p,q+N) || same(p,q+2*N))
{ans++;}
else
{
unite(p,q);
unite(p+N,q+N);
unite(p+2*N,q+2*N);

}
}

else
{
if( same(p,q) || same(p,q+2*N))
{ans++;}
else
{
unite(p,q+N);
unite(p+N,q+2*N);
unite(p+2*N,q);
}

}
}
else
ans++;
}
printf("%d\n",ans );
return 0;
}