我的并查集总结并记录

并查集模板

参考来源于并查集(Union-Find)算法介绍

• 并查集依据树的数据结构建立
• 并查集主要有两个基本操作
• 一是 find 算法（查找树的根结点）
• 二是 union算法（将两个不同根结点树连接在一起）
• 各类的并查集题目主要围绕着这两个操作进行更改变化

基本函数以及初始化（部分取舍根据题目改变）

#include <iostream>
using namespace std;
// 并查集
int id[80];			//代表总共有多少个结点
int size[180];		//代表每颗树的大小

//查找根结点
int find(int p)
{
	//通过循环更替查找根结点
	while (p != id[p])
	{
		//路径压缩，优化 find 的算法，会导致结点结构被破坏
		id[p] = id[id[p]];	
 		p = id[p];
	}
	return p;
}
//连接两个根结点
void Union(int p,int q)
{
	//获取两个结点的根结点
	int i = find(p);
	int j = find(q);
	//如果两个根结点相等，则说明两结点在同一个树下，不需要连接
	if(i == j) return;
//判断哪个树比较大，将小树连接到大树，实现对树的扁平化，达到优化效果
	if(size[i] < size[j])
	{
		id[i] = j;
		//更新连接后的树大小
		size[j] += size[i];
	}else
	{
		id[j] = i;
		size[i] += size[j];
	}
}
int main()
{
	//初始化每一个结点为一颗树，大小为 1
	for(int i = 0; i < 80; i++)
	{
		id[i] = i;	
		size[i] = 1;
	}
	system("pause");
	return 0;
}

求不同树的数量

//求一共有多少个不同的树
int getSum()
{
	int sum = 0;
	//遍历全部结点，如果该结点等于根结点，则说明是一个树
	for(int i = 0; i < 80; i++)
	{
		if(id[i] == i)
			sum++;
	}
	cout << sum ;
}