并查集-----

最新推荐文章于 2024-06-12 10:00:23 发布

1ts0k

最新推荐文章于 2024-06-12 10:00:23 发布

阅读量35

点赞数

文章标签：算法 c++ 数据结构 Powered by 金山文档

本文链接：https://blog.csdn.net/wetqadsg/article/details/129570680

版权

for (int i = 1; i <= n; i ++ ) p[i] = i;
 
int find(int x)
{
    if (p[x] != x) p[x] = find(p[x]);
// 路径压缩(并查集最重要的优化，让路径上的所有点都指向根节点，使得下次查找只需要o1的时间)
    return p[x];
}
 
p[find(a)] = find(b);
 
if (find(a) == find(b)) {...}

----------------------------------------------------------------------------

维护SIZE的并查集
int p[N], size[N];                  
//p[]存储每个点的祖宗节点, size[]只有祖宗节点的有意义，表示祖宗节点所在集合中的点的数量
 
int find(int x)
{
    if (p[x] != x) p[x] = find(p[x]);
    return p[x];
}
 
for (int i = 1; i <= n; i ++ )
{
    p[i] = i;
    size[i] = 1;                
}
 
// 合并a和b所在的两个集合：
int x = find(a), y = find(b);
if (x != y) {
    p[x] = y;
    size[y] += size[x];
}
 
//求a所在集合的长度
Size[find(a)]
 
if (find(a) == find(b)) {...}

-------------------------------------------------------------------------------
维护到祖宗节点距离的并查集

int p[N], d[N];
//p[]存储每个点的祖宗节点, d[x]存储x到p[x]的距离
 
int find(int x)
{
    if (p[x] != x)
    {
        int u = find(p[x]);         // 先更新d[p[x]]
        //该点到根节点的距离 = 该点到父节点的距离 + 父节点到根节点的距离
        d[x] += d[p[x]];            
        // d[x]一开始存的是到父节点的距离，通过路径压缩更新成每个点到根节点的距离
        p[x] = u;                   // 最后更新p[x]
    }
    return p[x];
}
 
for (int i = 1; i <= n; i ++ )
{
    p[i] = i;
    d[i] = 0;                       
}
 
// 合并a和b所在的两个集合：
p[find(a)] = find(b);
d[find(a)] = distance; // 根据具体问题，初始化find(a)的偏移量