并查集（用C/C++实现）

一、 支持的操作

1. 将两个集合合并
2. 询问两个元素是否在一个集合当中

 

二、 基本原理

每个集合用一棵树来表示。树根的编号就是整个集合的编号。每个节点存储它的父节点，p[x]表示x的父节点。

1. 判断树根方法：if( p[x] == x )

2. 求x的集合编号的方法：while( p[x] != x ) x = p[x];

优化：（路径压缩）x一步一步找到了根节点之后，让所有节点全部指向根节点。

3. 合并两个集合的方法：把其中一棵树的树根插入另一棵树的树根之下。px是x的集合编号，py是y的集合编号，p[x] = y.

#include<iostream> 
/*1~n的数字，最开始每个数各自在一个集合中。
“M a b”将编号为a和b的两个数所在的集合合并，如果两个数已经在同一个集合中，则忽略。
“Q a b”询问编号为a和b的两个数是否在同一个集合中*/
using namespace std;

const int N = 100010;

int n, m;
int p[N];

int find(int x)//返回x的根节点 + 路径压缩 
{
	if( p[x] != x) p[x] = find(p[x]); 
	return p[x]; 
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i ++) p[i] = i;
	
	while(m --)
	{
		char op[2];//读字符串时scanf会自动忽略空格和回车，过滤技巧 
		int a, b;
		scanf("%s%d%d", op, &a, &b);
		
		if(op[0] == 'M') p[find(a)] = find(b);
		else
		{
			if(find(a) == find(b)) puts("Yes");
			else puts("No");
		}
	}
	return 0; 
}

 

三、 变形（维护额外的信息）

#include<iostream> 
/*给定一个包含n个点的无向图，初始时图中没有边，现要进行m个操作
“C a b”在a和b之间连一条线，a和b可能相等
“Q1 a b”询问a和b是否在同一个连通块中，a和b可能相等
“Q2 a”询问a所在连通块中点的数量*/
using namespace std;

const int N = 100010;

int n, m;
int p[N], size[N];

int find(int x)//返回x的根节点 + 路径压缩 
{
	if( p[x] != x) p[x] = find(p[x]); 
	return p[x]; 
}

int main()
{
	scanf("%d%d", &n, &m);
	for(int i = 1; i <= n; i ++)
	{
		p[i] = i;
		size[i] = 1;//只维护每一个根节点的大小 
	} 
	
	while(m --)
	{
		char op[2];//读字符串时scanf会自动忽略空格和回车，过滤技巧 
		int a, b;
		scanf("%s", op);
		
		if(op[0] == 'C')
		{
			scanf("%d%d", &a, &b);
			if(find(a) == find(b)) continue; 
			size[find(b)] += size[find(a)];//根节点相加 
			p[find(a)] = find(b);
		} 
		else if(op[1] == '1')
		{
			scanf("%d%d", &a, &b);
			if(find(a) == find(b)) puts("Yes");
			else puts("No");
		}
		else
		{
			scanf("%d", &a);
			printf("%d\n", size[find(a)]);
		}
	}
	return 0; 
}