首页 > 代码库 > POJ1182 食物链 (hard)

POJ1182 食物链 (hard)

 

Description
动物王国中有三类动物A,B,C,这三类动物的食物链构成了有趣的环形。A吃B, B吃C,C吃A。
现有N个动物,以1-N编号。每个动物都是A,B,C中的一种,但是我们并不知道它到底是哪一种。
有人用两种说法对这N个动物所构成的食物链关系进行描述:
第一种说法是"1 X Y",表示X和Y是同类。
第二种说法是"2 X Y",表示X吃Y。
此人对N个动物,用上述两种说法,一句接一句地说出K句话,这K句话有的是真的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。
1) 当前的话与前面的某些真的话冲突,就是假话;
2) 当前的话中X或Y比N大,就是假话;
3) 当前的话表示X吃X,就是假话。
你的任务是根据给定的N(1 <= N <= 50,000)和K句话(0 <= K <= 100,000),输出假话的总数。 


思路:把一个点虚化成三个点,a(本身),a+n(a吃的),a+2*n(吃a的)。然后每读入一个1操作,就判断a、b间有无吃与被吃的关系,然后合并a和b、a+n和b+n、a+2*n和b+2*n;每读入一个2操作,就判断a、b是否是同类和b是否吃a,然后合并a+n和b、a+2*n和b+n、a和b+2*n。注意要判断2)、3)两个条件。

code:
 #include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int fa[150001]={0},eat[150001]={0},eaten[150001]={0};
int rool(int x)
{
if (x==0) return x; 
if (fa[x]!=x) fa[x]=rool(fa[x]);
return fa[x];
}
int main()
{
int n,k,i,j,ans=0,kind,x,y;
cin>>n>>k;
for (i=1;i<=n*3;++i)
  fa[i]=i;
for (i=1;i<=k;++i)
{
cin>>kind>>x>>y;
if (x>n||y>n) 
{
++ans;
continue;
}
if (kind==1)
{
if ((rool(x+n)==rool(y))||(rool(x+2*n)==rool(y))) 
{
++ans;
continue;
}
fa[rool(x)]=rool(y);
fa[rool(x+n)]=rool(y+n);
fa[rool(x+2*n)]=rool(y+2*n);
}
if (kind==2)
{
if (rool(x)==rool(y)||rool(x+2*n)==rool(y))
{
++ans;
continue;
}
fa[rool(x+n)]=rool(y);
fa[rool(x+2*n)]=rool(y+n);
fa[rool(x)]=rool(y+2*n);
}
}
cout<<ans<<endl;

 

POJ1182 食物链 (hard)