并查集基本代码实现

并查集基本代码实现

#include<cstdio>
#include<cstring>
#include<string>
#include<iostream>
#include<stack>
#include<algorithm>
#include<map>
#include<set>
#include<cmath>
#include<queue>
using namespace std;


const int MAX_N=10000;


int par[MAX_N];//父亲
int rank[MAX_N];//树的高度

//初始化n个元素
void init(int n)
{
	for (int i = 0; i < n; ++i)
	{
		par[i]=i;
		rank[i]=0;
	}
}

//查询树的根
int find(int x)
{
	if(par[x]==x)
	{
		return  x;
	}
	else
		return par[x]=find(par[x]);
}

//合并x和y所属的集合
void unite(int x,int y)
{
	x=find(x);
	y=find(y);
	if (x==y)
	{
		return;
	}

	if (rank[x]<rank[y])
	{
		par[x]=y;
	}
	else
	{
		par[y]=x;
		if (rank[x]==rank[y])
		{
			rank[x]++;
		}
	}
}

bool same(int x,int y)
{
	return find(x)==find(y);
}

int main()
{
	init(10);
	return 0;
}