您的位置:首页 > 其它

贴一个并查集的模板吧

2014-04-06 13:51 253 查看
实现并查集数据结构主要有三个函数。代码如下:

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[ ]值

    }

}
内容来自用户分享和网络整理,不保证内容的准确性,如有侵权内容,可联系管理员处理 点击这里给我发消息
标签: