洛谷P2024 食物链

题目描述

动物王国中有三类动物 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 句话有的是真

的,有的是假的。当一句话满足下列三条之一时,这句话就是假话,否则就是真话。

• 当前的话与前面的某些真的话冲突,就是假话

• 当前的话中 X 或 Y 比 N 大,就是假话

• 当前的话表示 X 吃 X,就是假话

你的任务是根据给定的 N 和 K 句话,输出假话的总数。

输入输出格式

输入格式:

从 eat.in 中输入数据

第一行两个整数,N,K,表示有 N 个动物,K 句话。

第二行开始每行一句话(按照题目要求,见样例)

输出格式:

输出到 eat.out 中

一行,一个整数,表示假话的总数。

输入输出样例

输入样例#1: 复制
100 7
1 101 1
2 1 2
2 2 3
2 3 3
1 1 3
2 3 1
1 5 5
输出样例#1: 复制
3

说明

1 ≤ N ≤ 5 ∗ 10^4

1 ≤ K ≤ 10^5


带权并查集,f[x][0]存x的父亲,f[x][1]存x与父亲的关系:0表示x与父亲是同类,1为x吃父亲,2为x的父亲吃x(...
可以再每次查询祖先的时候将f数组压缩
原理:将x到树顶的路径f[i][0]加和最后%3即为x与树根关系。
 
#include<iostream>
#include<stdio.h>

using namespace std;

int i,m,n,j,k,h,g,f[1000001][2],b,ans;

int father(int x)
{
	if(f[x][0]==x) 
	{
		m=f[x][1];
		return f[x][0];
	}
	f[x][0]=father(f[x][0]);
	f[x][1]=(f[x][1]+m+3)%3;
	m=f[x][1];
	return f[x][0];
}

int main()
{
	scanf("%d%d",&n,&k);
	for(i=1;i<=n;i++) f[i][0]=i;
	for(i=1;i<=k;i++) 
	{
		scanf("%d%d%d",&b,&g,&h);
		if((g>n || h>n)||(g==h && b==2)) 
		{
			ans+=1;
			continue;
		}
		m=0;
		father(g);
		m=0;
		father(h);
		if(f[g][0]==f[h][0])
		{
			if(b==1) if(f[g][1]!=f[h][1]) ans+=1;
			if(b==2) if((f[g][1]+2)%3!=f[h][1]) ans+=1;  
		}
		else 
		{
			if(b==1) 
			{
				f[f[g][0]][1]=(f[h][1]+3-f[g][1])%3;
				f[f[g][0]][0]=f[h][0];
			}
			if(b==2)
			{
				f[f[h][0]][1]=(f[g][1]+5-f[h][1])%3;
				f[f[h][0]][0]=f[g][0];
			}
		}
	}
	
	printf("%d",ans);
}

原文地址:https://www.cnblogs.com/ZUTTER/p/9289274.html