学习笔记：并查集

并查集是一种树型的数据结构，用于处理一些不相交集合的合并及查询问题。

AcWing 836. 合并集合

一共有n个数，编号是1~n，最开始每个数各自在一个集合中。
现在要进行m个操作，操作共有两种：
“M a b”，将编号为a和b的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略这个操作；
“Q a b”，询问编号为a和b的两个数是否在同一个集合中；
输入格式
第一行输入整数n和m。
接下来m行，每行包含一个操作指令，指令为“M a b”或“Q a b”中的一种。

输出格式
对于每个询问指令”Q a b”，都要输出一个结果，如果a和b在同一集合内，则输出“Yes”，否则输出“No”。每个结果占一行。（1≤n,m≤1e5）

const int N = 1e5+5;

int n,m,pre[N];

int find(int x)	{ return pre[x]==x?x:pre[x]=find(pre[x]); }//find函数，用与查找x所在集合根节点
void join(int x,int y)	{ pre[find(x)]=find(y); }//将x所在的集合与y所在的集合合并，并且一y的根节点为父亲

int main()
{
	cin>>n>>m;
	char op; int a,b;
	for(int i=1;i<=n;i++)	pre[i]=i; //并查集初始化，一开始每个人的父亲节点都是自己
	while(m--)
	{
		cin>>op>>a>>b;
		if(op=='M')
		{
			if(find(a)!=find(b))	join(a,b);
		}
		else
		{
			if(find(a)==find(b))	cout<<"Yes"<<endl;
			else	cout<<"No"<<endl;
		}
	}
	return 0;
}
	

并查集也可以维护一些信息，比如集合中元素的数量，集合中元素之间的关系等

AcWing 837. 连通块中点的数量

给定一个包含n个点（编号为1~n）的无向图，初始时图中没有边。
现在要进行m个操作，操作共有三种：
“C a b”，在点a和点b之间连一条边，a和b可能相等；
“Q1 a b”，询问点a和点b是否在同一个连通块中，a和b可能相等；
“Q2 a”，询问点a所在连通块中点的数量；
输入格式
第一行输入整数n和m。
接下来m行，每行包含一个操作指令，指令为“C a b”，“Q1 a b”或“Q2 a”中的一种。

输出格式
对于每个询问指令”Q1 a b”，如果a和b在同一个连通块中，则输出“Yes”，否则输出“No”。
对于每个询问指令“Q2 a”，输出一个整数表示点a所在连通块中点的数量每个结果占一行。（1≤n,m≤1e5）

const int N = 1e5+5;

int n,m,pre[N],sizes[N];

int find(int x)	{ return pre[x]==x?x:pre[x]=find(pre[x]); }
void join(int x,int y)
{
	int fx=find(x), fy=find(y);
	if(fx!=fy)	pre[fx]=fy,sizes[fy]+=sizes[fx]; //如果x与y不在一个集合中，合并x，y的集合，并更新sizes数组
}

int main()
{
	cin>>n>>m;
	string op; int a,b;
	for(int i=1;i<=n;i++)	pre[i]=i,sizes[i]=1; //并查集初始化，同时sizes数组也初始化，一开始每个点单独一个集合，每个集合中只有一个元素，即本身
	while(m--)
	{
		cin>>op;
		if(op=="C")
		{
			cin>>a>>b;
			join(a,b);	
		}
		else if(op=="Q1")
		{
			cin>>a>>b;
			if(find(a)==find(b))	cout<<"Yes"<<endl;
			else	cout<<"No"<<endl;
		}
		else
		{
			cin>>a;
			cout<<sizes[find(a)]<<endl;
		}
	}
	return 0;
}

AcWing 240. 食物链

动物王国中有三类动物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。
输出格式
只有一个整数，表示假话的数目。
（1≤N≤50000, 0≤K≤100000）

思路：
对于所有的动物，都把它放到一个集合中，但是由于3种动物之间的关系需要维护，所以要根据它们之间的某种联系吧这种关系能确定下来，考虑到根节点的特殊性，我们可以根据每个点与根节点之间的距离来确定每个节点与根节点的之间的关系，因为只有3种动物，我们可以用距离%3得到的余数来代表它们的关系，如果节点x到根节点的距离d[x]%3=0,表示x与根结点是同一种动物；如果d[x]%3=1，表示x是可以吃根节点的动物；如果d[x]%3=2,表示根节点可以吃x
也就是这样一种关系：1 -> 0 , 2 -> 1 , 0 -> 2

在判断两个动物x与y的关系时就可以这样判断：
首先，找到x的祖先节点fx，y的祖先节点fy

如果 fx==fy ，也就是x与y在同一个集合里
那么，如果x与y是同一种动物，那么它们到根节点的距离%3得到的余数是相同的也就是 （d[x]-d[y]）%3=0;
如果x吃y ，那么它们到根节点的距离%3得到的余数满足这样一种关系：（d[x]-d[y]-1）%3=0;
因为只存在： 0吃2 ，1吃0 ，2吃1
0-2=-2
1-0=1
2-1=1
可以看出（-2-1）%3=0；（1-1）%3=0；

如果fx！=fy
也就是x与y不在一个集合里，那么就可以根据题目告诉我们的关系进行合并
如果x与y是同一种动物
那么合并之后（fx认fy做父亲），(d[x]+d[fx]-d[y])%3应该是等于0的，但是在合并之前d[fx]=0; 所以由上面的式子可以得出d[fx]=d[y]-d[x];
如果x吃y
那么合并之后（fx认fy做父亲），（d[x]+d[fx]-d[y]-1）%3一个是等于0的，同样的合并之前d[fx]作为祖先节点是等于0的，所以由上面的式子可以得出d[fx]=d[y]-d[x]+1

const int N = 5e4+5;

int n,k,pre[N],d[N];
//与根节点的距离 %3 = 0,1,2 分为3类
// 0代表与根节点是同一类，1代表可以吃0,2可以吃1，同时0吃2 (0->2,2->1,1->0) 

int find(int x)
{
	if(pre[x]!=x) //这里在路径压缩时不要忘了维护d数组
	{
		int fx=find(pre[x]);
		d[x]+=d[pre[x]];
		pre[x]=fx;
	}
	return pre[x];
}

bool check(int id,int x,int y)
{
	if(x>n||y>n)	return false;//超出范围 
	
	int fx=find(x), fy=find(y);//先找到祖先节点
	if(id==1) // x与y是同类 
	{
		if(fx==fy)//x与y在一个集合中
		{
			if((d[x]-d[y])%3!=0)	return false;// 如果x与y在原先维护的d数组中并不是同一类
			else	return true;
		}
		//x与y不在一个集合中，那么根据关系进行合并 
		pre[fx]=fy;
		d[fx]=d[y]-d[x];
	}
	else //x吃y 
	{
		if(fx==fy)
		{
			if((d[x]-d[y]-1)%3!=0)	return false; // 如果x与y在原先维护的d数组中并不是x吃y的关系
			else	return true;
		}
		//x与y不在一个集合中，那么根据关系进行合并 
		pre[fx]=fy;
		d[fx]=d[y]-d[x]+1;
	}
	return true;
}

int main()
{
	cin>>n>>k;
	for(int i=1;i<=n;i++)	pre[i]=i;
	
	int cnt=0,id,x,y;
	while(k--)
	{
		cin>>id>>x>>y;
		if(!check(id,x,y))	cnt++; //如果是假话
	}
	cout<<cnt<<endl;
	return 0;
}