代码引自挑战程序设计竞赛
int par[MAX_N]; //代表某个节点的父节点
int rank[MAX_N]; //代表某个节点的节点高度
void init(int n) //初始化n个节点的值
{
for(int i=0;i<n;i++)
{
par[i]=i;
rank[i]=0;
}
}
int find(int x) //查询索引为x的节点的父节点,顺便更新x的父节点
{
if(par[x]==x)
{
return x;
}
else
return par[x]=find(par[x]);
}
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);
}