并查集

目的:实现对元素的分组与查找
代码实现:

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]++;
		}
	}
}

//判断x与y是否属于同一个集合
bool same(int x,int y)
{
	return find(x)==find(y);
}
posted @ 2021-12-09 15:33  Lord-Phantom-city  阅读(32)  评论(0)    收藏  举报