1.26寒假集训(并查集三连←.←)-4

C - 食物链

动物王国中有三类动物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
100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5
Sample Output
3


题意:一开始的写法令人智熄的TLE了,于是去网上参考了一下大佬的做法,将每个小动物置三种形态,太强了ORZ


#include<cstdio>
#include<cmath>
#include<algorithm>
#include<string>
#include<iostream>
using namespace std;

int fa[155555];

int findfa(int x)//找爸爸
{
	int a=x;
	while(x!=fa[x])
	{
		x=fa[x];
	}
	while(a!=fa[a])
	{
		int z=a;
		a=fa[a];
		fa[z]=x;
	}
	return x;
}


void uni(int a,int b)//合并两个小朋友
{
	int faa=findfa(a);
	int fab=findfa(b);
	if(faa!=fab)
	{
		fa[faa]=fab;
	}
}

int N,K,D,X,Y,ans;

int main()
{
	scanf("%d %d",&N,&K);
	for(int i=1;i<=3*N;i++)//每个动物有三种可能状态
		fa[i]=i;
	for(int i=1;i<=K;i++)
	{
		scanf("%d %d %d",&D,&X,&Y);
		if(X>N||Y>N)
		{
			ans++;
			continue;
		}
		if(D==1)
		{
			if(findfa(X)==findfa(Y+N)||findfa(X)==findfa(Y+2*N))//二者为捕食关系,不是同类的情况
			{
				ans++;
			}
			else
			{
				uni(X,Y);
				uni(X+N,Y+N);
				uni(X+2*N,Y+2*N);//将二者置为同类关系
			}
		}
		else
		{
			if(findfa(X)==findfa(Y)||findfa(X+2*N)==findfa(Y+N)||findfa(X)==findfa(Y+2*N))//二者为同类或者后者捕食前者
			{
				ans++;
			}
			else
			{
				uni(X,Y+N);
				uni(X+N,Y+2*N);
				uni(X+2*N,Y);//将二者置为前者捕食后者的关系
			}
		}
	}
	printf("%d\n",ans);
	return 0;
}


猜你喜欢

转载自blog.csdn.net/qq_39396954/article/details/79172512