并查集（偏移量+路径压缩）

并查集的概念：

并查集是由一个数组、两个函数（查找一个数的根、合并路线）组成，常用来解决一些不相交集合的合并与查找问题。

并查集中有2个重要的函数：一个用来寻找祖先，一个用来合并

祖先：节点指向自身

int find(int x) //寻找x的祖先，这里为了节约时间使用了路径压缩
{
	return x == p[x] ? x : p[x] = find(p[x]);
}

路径压缩：

下图root为祖先，下面依次是他的儿子节点，不难发现下面三个节点的祖先都是root（图里的应该是r1，r2，r3，打错了，懒得改了= =领会精神哈），那么我们可以让这个三个节点都直接指向他们的祖先root
下面是压缩后的路径，这样就可以大大减少查找祖先的时间

下面是一道并查集的模板题：

亲戚(relation)

Description

或许你并不知道，你的某个朋友是你的亲戚。他可能是你的曾祖父的外公的女婿的外甥女的表姐的孙子。如果能得到完整的家谱，判断两个人是否亲戚应该是可行的，但如果两个人的最近公共祖先与他们相隔好几代，使得家谱十分庞大，那么检验亲戚关系实非人力所能及。在这种情况下，最好的帮手就是计算机。为了将问题简化，你将得到一些亲戚关系的信息，如Marry和Tom是亲戚，Tom和Ben是亲戚，等等。从这些信息中，你可以推出Marry和Ben是亲戚。请写一个程序，对于我们的关于亲戚关系的提问，以最快的速度给出答案。

Input

输入由两部分组成。

第一部分以N，M开始。N为问题涉及的人的个数(1≤N≤20000)。这些人的编号为1,2,3,…, N。下面有M行(1≤M≤1 000 000)，每行有两个数ai, bi，表示已知ai和bi是亲戚。

第二部分以Q开始。以下Q行有Q个询问(1≤Q≤1 000 000)，每行为ci, di，表示询问ci和di是否为亲戚。

Output

对于每个询问ci, di，输出一行：若ci和di为亲戚，则输出“Yes”，否则输出“No”。

Sample Input 1

10 7
2 4
5 7
1 3
8 9
1 2
5 6
2 3
3
3 4
7 10
8 9

Sample Output 1

Yes
No
Yes

#include <bits/stdc++.h>

using namespace std;

const int maxn = 2e5+5;
int p[maxn];
int m,n;

int find(int x)	//这是迭代版本的find函数，使用迭代不容易发生错误
{
	int k, j, r;
    r = x;
	while(r != p[r])  
        r = p[r];      
    k = x;        
	while(k != r)             
    {
        j = p[k];         
        p[k] = r;        
        k = j;                    
    }
	return r;                  
}

int main()
{
	scanf("%d %d",&n,&m);
	for(int i = 1 ; i <= 2*n ; i++)
		p[i] = i;
	for(int i = 0 ; i < m ; i++)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		int r1 = find(a),r2 = find(b);	//找到a，b的祖先
		p[r1] = r2;	
	}
	int q;
	scanf("%d",&q);
	while(q--)
	{
		int a,b;
		scanf("%d %d",&a,&b);
		if(find(a) == find(b))
			printf("Yes\n");
		else
			printf("No\n");
	}
	return 0;
}

在一般的并查集的基础上就是带权并查集了，带权并查集比较难理解，因为其中涉及到一个偏移量的问题，移量的计算与向量有关

如图dis[a]表示a到fa的偏移量，如果我们想计算fa到fb的偏移量也就是要求出dis[fa]，这里不能用一般的加减法，要使用向量的加减，也就是c+dis[b]-dis[a]，这样就算出了dis[fa]，下面是一题带权并查集的应用

食物链（eat）

Description

动物王国中有三类动物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 1

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 1

3

#include<bits/stdc++.h>

using namespace std;

const int maxn = 1e5+5;
const int mod = 3;
int p[maxn];
int dis[maxn];

int find(int x)
{
	if(x != p[x])
	{
		int root = p[x];
		p[x] = find(p[x]);
		dis[x] = (dis[x] + dis[root]) % mod;	
	}	
	return p[x];
}

int main()
{
	int n,k;
	cin >> n >> k;
	for(int i = 1 ; i <= n ; i++)
		p[i] = i;
	int ans = 0;
	for(int i = 0 ; i < k ; i++)
	{
		int d,x,y;
		cin >> d >> x >> y;
		if(x > n || y > n)
		{
			ans++;
			continue;
		}
		int r1 = find(x),r2 = find(y);
		if(r1 != r2)
		{
			p[r1] = r2;
			dis[r1] = (dis[y] - dis[x] + d - 1 + mod) % mod;
		}
		else
		{
			if((dis[x] - dis[y] + mod) % mod != d - 1)
				ans++;
		}
	}
	cout << ans << endl;
	return 0;
} 

在一维的基础上演变出二维并查集

格子游戏（grid）

Description

Alice和Bob玩了一个古老的游戏：首先画一个n * n的点阵（下图n = 3）

接着，他们两个轮流在相邻的点之间画上红边和蓝边：

直到围成一个封闭的圈（面积不必为1）为止，“封圈”的那个人就是赢家。因为棋盘实在是太大了(n <= 200)，他们的游戏实在是太长了！他们甚至在游戏中都不知道谁赢得了游戏。于是请你写一个程序，帮助他们计算他们是否结束了游戏？

Input

输入数据第一行为两个整数n和m。m表示一共画了m条线。以后m行，每行首先有两个数字(x, y)，代表了画线的起点坐标，接着用空格隔开一个字符，假如字符是"D "，则是向下连一条边，如果是"R "就是向右连一条边。输入数据不会有重复的边且保证正确。

Output

输出一行：在第几步的时候结束。假如m步之后也没有结束，则输出一行“draw”。

Sample Input 1

3 5
1 1 D
1 1 R
1 2 D
2 1 R
2 2 D

Sample Output 1

4

#include <bits/stdc++.h>

using namespace std;

const int maxn = 1e3+5;

struct node
{
	int x,y;
}p[maxn][maxn],k1,k2;


node find(node t)
{
	if(t.x == p[t.x][t.y].x && t.y == p[t.x][t.y].y)
		return t;
	else
		return p[t.x][t.y] = find(p[t.x][t.y]);
}

int main()
{
	int n,m;
	int ans = 0;
	cin >> n >> m;
	for(int i = 1 ; i <= n ; i++)
	{
		for(int j = 1 ; j <= n ; j++)
		{
			p[i][j].x = i;
			p[i][j].y = j;
		}
	}
	for(int i = 0 ; i < m ; i++)
	{
		int x,y;
		char c;
		cin >> x >> y >> c;
		if(c == 'D')
		{
			k1 = find(p[x][y]);
			k2 = find(p[x+1][y]);
		}
		else
		{
			k1 = find(p[x][y]);
			k2 = find(p[x][y+1]);
		}
		if(k1.x == k2.x && k1.y == k2.y)
		{
			cout << i + 1 << endl;
			return 0;
		}
		else
		{
			p[k2.x][k2.y] = k1;
		}
	}
	cout << "draw" << endl;
	return 0;
}