POJ 1182 食物链-优快云博客

食物链

Time Limit: 1000MS		Memory Limit: 10000K
Total Submissions: 30668		Accepted: 8944

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），输出假话的总数。

Input

第一行是两个整数N和K，以一个空格分隔。
以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。
若D=1，则表示X和Y是同类。
若D=2，则表示X吃Y。

Output

只有一个整数，表示假话的数目。

Sample Input

Sample Output

Source

Noi 01

当初想了两天写了个代码然后WA了就以为自己的算法错了然后过了N久想起来去看了眼discuss 有人说把while(~scanf())改成单组输入就过了~ 就把曾经WA的代码改了试了一下，结果AC了。。

#include<iostream>
#include<cstdio>
#include<cstring>
using namespace std;

int p[50005];
int a1[50005];//a1[i]表示i动物集合可以吃的动物所在集合
int a2[50005];//a2[i]表示吃i动物集合的动物所在集合
int rank[50005];

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

void unio(int x,int y)
{
	int a=find(x);
	int b=find(y);
	if(a==b)return;
	if(rank[a]<rank[b])
	{
		p[a]=b;
	}
	else if(rank[b]<rank[a])
	{
		p[b]=a;
	}
	else
	{
		p[a]=b;
		rank[b]++;
	}
}

void init_set(int n)
{
	for(int i=1;i<=n;i++)
	{
		p[i]=i;
		rank[i]=1;
		a1[i]=-1;
		a2[i]=-1;
	}
}

int main()
{
	int n,k;
	while(~scanf("%d%d",&n,&k))
	{
		init_set(n);
		int res=0;

		for(int i=0;i<k;i++)
		{
			int cmd,x,y;
			scanf("%d%d%d",&cmd,&x,&y);
			if(x>n||y>n)//出现大于n的动物 假话
			{
				res++;
				continue;
			}
			if(cmd==1)
			{
				if(x==y)continue;
				int s_x=find(x);
				int s_y=find(y);
				if(s_x==s_y)
				{
					continue;
				}
				if(a1[s_x]==s_y||a2[s_x]==s_y)//出现茅盾，s_y动物集合是吃s_x集合的动物或者是被s_x吃的，不相等
				{
					res++;
					continue;
				}
				int eat_x=a2[s_x];//eat_x是吃s_x集合的动物集合
				int x_eat=a1[s_x];//s_x集合可以吃的动物集合
				int eat_y=a2[s_y];
				int y_eat=a1[s_y];
				unio(s_x,s_y);//s_x,s_y 合并
				int s1=find(s_x);
				if(eat_x!=-1&&eat_y!=-1)//吃s_x的和吃s_y的合并
				{
					unio(eat_x,eat_y);
				}
				if(x_eat!=-1&&y_eat!=-1)//被s_x吃的和被s_y吃的合并
				{
					unio(x_eat,y_eat);
				}
				int s2=-1;//将s2赋为吃x和y的集合
				if(eat_x!=-1&&eat_y!=-1)
				{
					s2=find(eat_x);
				}
				else if(eat_x!=-1)
				{
					s2=eat_x;
				}
				else if(eat_y!=-1)
				{
					s2=eat_y;
				}
				int s3=-1;//s3赋为被x和y吃的
				if(x_eat!=-1&&y_eat!=-1)
				{
					s3=find(x_eat);
				}
				else if(x_eat!=-1)
				{
					s3=x_eat;
				}
				else if(y_eat!=-1)
				{
					s3=y_eat;
				}
				if(s2!=-1)//修正s1 s2 s3之间的吃与被叫关系
				{
					a1[s2]=s1;
					a2[s1]=s2;
				}
				if(s3!=-1)
				{
					a1[s1]=s3;
					a2[s3]=s1;
				}
				if(s2!=-1&&s3!=-1)
				{
					a1[s3]=s2;
					a2[s2]=s3;
				}
			}
			else if(cmd==2)
			{
				if(x==y)
				{
					res++;
					continue;
				}
				int s_x=find(x);
				int s_y=find(y);
				if(a2[s_x]==s_y||s_x==s_y)
				{
					res++;
					continue;
				}
				int x_eat=a1[s_x];
				int eat_x=a2[s_x];

				int y_eat=a1[s_y];
				int eat_y=a2[s_y];

				if(eat_y!=-1)
				{
					unio(s_x,eat_y);
				}
				if(x_eat!=-1)
				{
					unio(x_eat,s_y);
				}
				if(eat_x!=-1&&y_eat!=-1)
				{
					unio(eat_x,y_eat);
				}
				int s1=find(s_x);
				int s2=find(s_y);
				a1[s1]=s2;
				a2[s2]=s1;
				int s3=-1;
				if(eat_x!=-1&&y_eat!=-1)
				{
					s3=find(eat_x);
				}
				else if(eat_x!=-1)
				{
					s3=eat_x;
				}
				else if(y_eat!=-1)
				{
					s3=y_eat;
				}
				if(s3!=-1)
				{
					a1[s3]=s1;
					a2[s1]=s3;
					a1[s2]=s3;
					a2[s3]=s2;
				}
			}
		}
		printf("%d\n",res);
	}
	return 0;
}