实现并查集数据结构主要有三个函数。代码如下:
void UFset( ) //初始化
{
for( int i=0; i<N; i++ )
parent[i] = -1;
}
int Find( int x ) //查找并返回结点x 所属集合的根结点
{
int s; //查找位置
//一直查找到parent[s]为负数(此时的s 即为根结点)为止
for( s=x; parent[s]>=0; s=parent[s] )
;
while( s!=x ) //优化方案―压缩路径,使后续的查找操作加速
{
int tmp = parent[x];
parent[x] = s;
x = tmp;
}
return s;
}
//R1 和R2 是两个元素,属于两个不同的集合,现在合并这两个集合
void Union( int R1, int R2 )
{
//r1 为R1 的根结点,r2 为R2 的根结点
int r1 = Find(R1), r2 = Find(R2);
int tmp = parent[r1] + parent[r2]; //两个集合结点个数之和(负数)
//如果R2 所在树结点个数 > R1 所在树结点个数
//注意parent[r1]和parent[r2]都是负数
if( parent[r1] > parent[r2] ) //优化方案――加权法则
{
parent[r1] = r2; //将根结点r1 所在的树作为r2 的子树(合并)
parent[r2] = tmp; //更新根结点r2 的parent[ ]值
}
else
{
parent[r2] = r1; //将根结点r2 所在的树作为r1 的子树(合并)
parent[r1] = tmp; //更新根结点r1 的parent[ ]值
}
}
void UFset( ) //初始化
{
for( int i=0; i<N; i++ )
parent[i] = -1;
}
int Find( int x ) //查找并返回结点x 所属集合的根结点
{
int s; //查找位置
//一直查找到parent[s]为负数(此时的s 即为根结点)为止
for( s=x; parent[s]>=0; s=parent[s] )
;
while( s!=x ) //优化方案―压缩路径,使后续的查找操作加速
{
int tmp = parent[x];
parent[x] = s;
x = tmp;
}
return s;
}
//R1 和R2 是两个元素,属于两个不同的集合,现在合并这两个集合
void Union( int R1, int R2 )
{
//r1 为R1 的根结点,r2 为R2 的根结点
int r1 = Find(R1), r2 = Find(R2);
int tmp = parent[r1] + parent[r2]; //两个集合结点个数之和(负数)
//如果R2 所在树结点个数 > R1 所在树结点个数
//注意parent[r1]和parent[r2]都是负数
if( parent[r1] > parent[r2] ) //优化方案――加权法则
{
parent[r1] = r2; //将根结点r1 所在的树作为r2 的子树(合并)
parent[r2] = tmp; //更新根结点r2 的parent[ ]值
}
else
{
parent[r2] = r1; //将根结点r2 所在的树作为r1 的子树(合并)
parent[r1] = tmp; //更新根结点r1 的parent[ ]值
}
}