并查集简单模板

并查集解决的问题很典型，就是等价类划分的问题。

三个函数：

void init(int *p, int n)
{
	for(int i=0; i<n; i++)
		p[i] = -1;
}

int Find(int x, int *pn)
{
	if(pn[x]<0)
		return x;
	return pn[x] = Find(pn[x], pn);
}

// 合并：大的合并到小的那里
void Union(int x, int y, int *pn)
{
	int h1 = Find(x, pn);
	int h2 = Find(y, pn);
	if(h1<h2)
	{
		pn[h2] = h1;
		pn[h1]--;
	}
	else if(h1==h2) // 同属一个等价类，则返回
	{
		return;
	}
	else{pn[h1] = h2;pn[h2]--;}}

给定n个元素，m个等价关系，要你去划分出等价类，求等价类的个数，和每个等价类包含多少个元素。

【示例1】

// 并查集示例：合并等价类，最后可以计算：
// 1.共有几个等价类;
// 2.每个等价类包含多少个元素;
// 3.给定两个元素，是否属于同一等价类
// 示例输入：
//12 8
//0 2
//2 4
//1 7
//3 5
//4 8
//10 11
//0 6
//1 5

#include <stdio.h>
#include <iostream>
using namespace std;

void init(int *p, int n)
{
	for(int i=0; i<n; i++)
		p[i] = -1;
}

int Find(int x, int *pn)
{
	if(pn[x]<0)
		return x;
	return pn[x] = Find(pn[x], pn);
}

// 合并：大的合并到小的那里
void Union(int x, int y, int *pn)
{
	int h1 = Find(x, pn);
	int h2 = Find(y, pn);
	if(h1<h2)
	{
		pn[h2] = h1;
		pn[h1]--;
	}
	else if(h1==h2) // 同属一个等价类，则返回
	{
		return;
	}
	else{pn[h1] = h2;pn[h2]--;}}int main(){int n;cin>>n; // 元素个数int *pn = new int [n];//int m;cin>>m; // 关系个数int i;init(pn, n); // 初始化为-1for( i=0; i<m; i++){int x,y;cin>>x>>y;Union(x,y,pn); // 合并}// 输出int count = 0;for(i=0; i<n; i++){if(pn[i]<0){++count;printf("第 %d 个集合 - 元素个数 %d\n", count, -pn[i]);}}printf("共计: %d 个集合\n", count);delete []pn;return 0;}

输出：
第 1 个集合 - 元素个数 5
第 2 个集合 - 元素个数 3
第 3 个集合 - 元素个数 1
第 4 个集合 - 元素个数 2
共计: 4 个集合

【示例2】

如图，拥有3个8连块，size分别是1,3,7.这个问题可以用搜索来做，也可以用并查集来做。

并查集空间O(n),时间O(n)。n为矩阵元素个数。代码：

#include <stdio.h>
#include <iostream>
using namespace std;

void init(int *p, int n)
{
	for(int i=0; i<n; i++)
		p[i] = -1;
}

int Find(int x, int *pn)
{
	if(pn[x]<0)
		return x;
	return pn[x] = Find(pn[x], pn);
}

// 合并：大的合并到小的那里
void Union(int x, int y, int *pn)
{
	int h1 = Find(x, pn);
	int h2 = Find(y, pn);
	if(h1<h2)    // 合并到小的那里
	{
		pn[h2] = h1;
		pn[h1]--;
	}
	else if(h1==h2) // 同属一个等价类，则返回
	{
		return;
	}
	else
	{
		pn[h1] = h2;
		pn[h2]--;
	}
}

int main()
{
	
	int m, n;
	cin>>m>>n; // m*n
	int *parent = new int [m*n];
	//
	int *sq = new int[m*n];  // 存放矩阵本身
	int i,j;
	init(parent, m*n); // 初始化为-1
	for( i=0; i<m; i++)
	{
		for(j=0; j<n; j++)
		{
			char ch;
			cin>>ch;  //scanf("%c ", &ch); // 要用cin，用scanf会出错
			sq[i*n + j] = ch - '0';
		}
	}

	// 打印矩阵
	for( i=0; i<m; i++)
	{
		for(j=0; j<n; j++)
		{
			printf("%d ",sq[i*n + j]);
		
		}
		printf("\n");
	}

	// 开始合并
	for( i=0; i<m; i++)
	{
		for(j=0; j<n; j++)
		{ // 四个方向：右方，右下，下方，左下。每个点都检查这四个点即可保证没有遗漏。

			if (j+1<n  &&  sq[i*n + j] == 1 && sq[i*n + j + 1] ==1) 
				Union(i*n+j, i*n+j+1, parent);
			if (i+1<m && j+1<n  &&  sq[i*n + j] == 1 && sq[(i+1)*n + j + 1] ==1) 
				Union(i*n+j, (i+1)*n+j+1, parent);
			if (i+1<m  &&  sq[i*n + j] == 1 && sq[(i+1)*n + j] ==1) 
				Union(i*n+j, (i+1)*n+j, parent);
			if (i+1<m && j-1>=0  && sq[i*n + j] == 1 &&  sq[(i+1)*n + j -1] ==1) 
				Union(i*n+j, (i+1)*n+j-1, parent);
		}
	}

	// 输出
	int count = 0;
	for(i=0; i<m*n; i++)
	{
		if(parent[i]<0 && sq[i] == 1)
		{
			++count;
			printf("第 %d 个黑格子集合 - 元素个数 %d - ta的根位于第 %d 个元素\n", count, -parent[i], i);
		}
	}
	printf("共计: %d 个集合\n", count);
	delete []parent;
	delete []sq;
	return 0;
}

输出：

1 0 0 1 0 0 
0 0 1 0 1 0 
0 0 0 0 0 0 
1 1 0 0 0 0 
1 1 1 0 0 0 
0 1 0 1 0 0 
第 1 个黑格子集合 - 元素个数 1 - ta的根位于第 0 个元素
第 2 个黑格子集合 - 元素个数 3 - ta的根位于第 3 个元素
第 3 个黑格子集合 - 元素个数 7 - ta的根位于第 18 个元素
共计: 3 个集合