【SSL 1679】食物链(并查集)

题目大意

 动物王国中有三类动物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和K句话，输出假话的总数。

输入格式

第一行是两个整数N和K，以一个空格分隔。

以下K行每行是三个正整数 D，X，Y，两数之间用一个空格隔开，其中D表示说法的种类。

若D=1，则表示X和Y是同类。

若D=2，则表示X吃Y。

输出格式

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

输入样例

100 7
1 101 1 
2 1 2
2 2 3 
2 3 3 
1 1 3 
2 3 1 
1 5 5

输出样例

3

基本思路

这题有两种做法，一种是带权并查集，一种是扩展域并查集。扩展域我们改日再讲，我们来看看带权并查集的做法。

我们把所有正确关系放入集合,记录每个动物与根的关系。

怎么记录呢？

每个动物与根的关系只有3个：

（1）吃根  用1表示；        （2）被根吃 用2表示；        （3）与根是同类 用0表示

查询时我们将其与根节点的高度差模3，若余数为0，则这个动物与根是同类。高度差模3为1时，有吃与被吃的关系。

我们怎么看：x吃y，x与y间的边权为1，y被z吃，y与z间的边权为2，（1+2）%3=0

另一种：x吃z，x与z边权为1，y被z吃，y与z边权为2，（（1-2）%3+3）%3=1

但在写代码的时候，我们在处理输入数据时只需处理与根是同类和吃根的情况。

假如我们现在要合并x，y，x吃y，那如何连接他们的根fx，fy呢？

我们看下面这个图：

另外说一下，这是一个闭环模型，像石头剪刀布那样 ，给出任意两个就可以知道他们的关系，就像上图，y吃o，o吃fy，我们自然就知道y被fy吃，此时d[y]=2，符合条件。

我们知道x吃y，即合并后（（d[x]-d[y]）%3+3）%3==1，合并后的d[x]是现有的d[x]加上fx到fy的距离，因为根变成fy了，fx也要并入fy，我们设fx到fy的距离是？，我们可以得到d[x]+?-d[y]=1，移项得?=d[y]-d[x]+1，如此可以得到fx到fy的距离。

核心代码

#include<bits/stdc++.h>
using namespace std;
typedef long long ll;
const int N=5e4+10;
int n,k,fa[N],d[N],t,x,y,ans;
//d[i]表示i与根的距离
inline int find(int p){//带边权并查集
	if(fa[p]!=p){//差不多是个记搜
		int tmp=fa[p];
		fa[p]=find(fa[p]);
		//递归回来统计
		d[p]+=d[tmp];
	}
	return fa[p];
}
int main(){
	ios::sync_with_stdio(false);
	cin>>n>>k;
	for(int i=1;i<=n;i++) fa[i]=i;
	while(k--){
		cin>>t>>x>>y;
		if(x>n||y>n){
			ans++;
			continue;
		}
		int fx=find(x),fy=find(y);
		if(t==1){//x y是同类 
			if(fx==fy&&(d[x]-d[y])%3!=0) ans++;
			else if(fx!=fy){
				fa[fx]=fy;d[fx]=d[y]-d[x];
			}
		}
		else{
			if(x==y){
				ans++;
				continue;
			}
			if(fx==fy&&((d[x]-d[y])%3+3)%3!=1) ans++;
			else if(fx!=fy){
				fa[fx]=fy;d[fx]=d[y]-d[x]+1;
				//fx认fy作父亲
				//d[x]+?-d[y]=1 要等于1才有吃与被吃的关系
				//?=d[y]-d[x]+1
			}
		}
	}
	cout<<ans;
	return 0;
}