并查集【原理及模板】

整理自紫书

1.又来管理元素分组情况的数据结构

2.可以合并、无法分割

3.使用树形结构实现（不是二叉树）

分组：                                                                                 对应的树：

                    

 

 ①初始化：n个节点表示n个元素，没有边

②合并：从一个组的根向另一个组的根连边，两棵树→一棵树，两组→一组

③查询：查询两个节点是否属于同一组→树的根是否为同一个

   ②、⑤的根为①。   ⑦的根为⑥。              ②、⑤同组。                     ⑦、②不同组。

4、并查集实现中的注意点

避免退化：

① 对于每棵树，记录这棵树的高度（rank）。

②合并时如果两棵树的rank不同，从小的向大的连边。

路径压缩（使并查集高效）

（1）对于每一个节点，一旦向上走到了一次根节点，就把这个点到父亲的边改为直接连向根。

⑦的根为⑥        →→    直接连到⑥上。

（2）不仅是所查询的节点，在查询过程中，向上经过的所有节点，都改为直接连到根上。

如果查询⑤，就可以知道②~⑤所有的节点的根都是①。

为了简单，即使根的高度发生了变化，也不修改rank。

5、并查集的复杂程度

对n个元素的并查集进行一次操作的复杂程度是O(α(n))，α(n)是阿克曼函数的反函数，比O(log(n))还要快，“均摊复杂度”。

6、并查集的实现

数组par表示父亲的编号，par[x] = x时，x是所在树的根（用编号表示元素）

int par[MAX_N];  //父亲
int rank[MAX_N];  //树的高度


//初始化n个元素
void init(int n)
{
    for(int i=0; i<n; i++)
    {
        par[i] = i;
        rank[i] = 0;
    }
}


//查询树的根
int find(int x)
{
    if(par[x] == x)
        return x;
    else
    {
        return par[x] = find(par[x]);
    }
}

//合并x和y所属的集合
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]++;
    }
}


//判断x和y是否属于同一集合
bool same(int x, int y)
{
    return find(x)==find(y);
}