P2024 [NOI2001] 食物链

题目：

# [NOI2001] 食物链[NOI2001] 食物链 - 洛谷

## 题目描述

动物王国中有三类动物 A,B,C，这三类动物的食物链构成了有趣的环形。A 吃 B，B 吃 C，C 吃 A。

现有 N 个动物，以 1 \sim 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$ 句话，输出假话的总数。

## 输入格式

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

第二行开始每行一句话（按照题目要求，见样例）

## 输出格式

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

## 样例 #1

### 样例输入 #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\le N\le 5 \times 10^4$，$1\le K \le 10^5$。

思路：带权并查集

设 0 为同类，1 为 捕食 ， 2 为天敌

合并：将两条食物链连接

void add(int a,int b,int link)//link是a与b的关系
{
	int fa=find(a);
	int fb=find(b);
	pre[fa]=fb;
	re[fa]=(link+re[b]-re[a]+3)%3;
}

查找：查找某点动物与根节点动物关系

int find(int x)
{
	if(x==pre[x]) 
	return x;
	int fx=pre[x];
	pre[x]=find(fx);       //这两行不能颠倒
	re[x]=(re[x]+re[fx])%3;//必须先更新距离根节点最近的位置
	return pre[x];

代码如下：

#include<bits/stdc++.h>
using namespace std;
const int N=1e5+10;
int n,m,sum=0;
int pre[N],re[N];
void init()
{
	for(int i=1;i<=n;i++)
	{
		pre[i]=i;
		re[i]=0;
	} 
}
int find(int x)
{
	if(x==pre[x]) 
	return x;
	int fx=pre[x];
	pre[x]=find(fx);
	re[x]=(re[x]+re[fx])%3;
	return pre[x];
	
}
void add(int a,int b,int link)//link是a与b的关系
{
	int fa=find(a);
	int fb=find(b);
	pre[fa]=fb;
	re[fa]=(link+re[b]-re[a]+3)%3;
}
int main()
{
	cin>>n>>m;
	init();
	while(m--)
	{
		int check,a,b;
		cin>>check>>a>>b;
		if(a<1||b<1||a>n||b>n||(check==2&&a==b))
		{
			sum++;
			continue;
		}
		if(check==1)//a与b同类
		{
			if(find(a)!=find(b)) //a与b不在同一树上
			add(a,b,0);          //合并到同一树上
			else if(find(a)==find(b)&&re[a]!=re[b])//a与b的关系不同
			sum++;
		}
		else if(check==2)//a吃b
		{
			if(find(a)!=find(b))//a与b不在同一树上
			add(a,b,1);         //合并到同一树上
			else if(find(a)==find(b))
			{
				int temp=(re[a]-re[b]+3)%3;//通过两者与根节点的关系判断
				if(temp!=1) sum++;
			}
		}
	}
	cout<<sum<<endl;
	return 0;
}