优化1:将遍历过的结点都改成树根的儿子
优化2:记录树的高度,将小树连向大树
//并查集
int par[MAX], rak[MAX]; //父结点, 树的高度
//初始化n个元素
void init(int n)
{
for(int i = 0; i < n; i++)
{
par[i] = i; rak[i] = 0;
}
}
//查询树的根
int find(int x)
{
return par[x] == x ? x : par[x] = find(par[x]); //优化1:将遍历过的结点都改成树根的儿子
}
//合并x,y所属集合
void unite(int x, int y)
{
x = find(par[x]); y = find(par[x]);
if(x == y) return ;
if(rak[x] < rak[y]) //优化2:将小树连向大树
par[x] = y;
else
{
par[y] = x;
if(rak[x] == rak[y]) rak[x]++; //更新树的高度
}
}
//判断x、y是否在同一集合
int same(int x, int y)
{
return find(x) == find(y);
}