数据结构-并查集

并查集

基础模板

// 初始化
void init(int n)
{  
    for (int i = 0 ; i < n; i ++)
    {
        p[i] = i ; 
    }
}
// 返回x所在集合的编号+顺便路径压缩
int find_root(int x)
{
    //if (p[x] != x) p[x] = find_root(p[x]) ;
    if (p[x] != x) {
        return find_root(p[x]) ;
    }
    return p[x] ;   
}
// 合并两个集合,将j合并到i中去
void merge(int i , int j)
{
    p[find_root(j)] = find_root(i);
}

模板优化

// 初始化
void init()
{
    for (int i = 1 ; i <= N ; ++i)
    {
        s[i] = i ;
        height[i] = 0 ;
    }
}
void add(int x  , int y)
{
    x = find_par(x) ;
    y = find_par(y) ;
    // 相等，就任意选一个合并到另一个，高度+1
    if (height[x] == height[y]) 
    {
        s[x] = y ;
        height[x] = height[y] + 1;
    } else {
    	// 如果小于，则让高度小的合并到高度大的
        if (height[x] < height[y]) {
            s[x] = y ;
        } else {
            s[y] = x ;
        }
    }
}

路径压缩

// 返回x所在集合的编号+顺便路径压缩
int find_root(int x)
{
    if (p[x] != x) p[x] = find_root(p[x]) ;
    return p[x] ;   
}
/*
	优化了下一次的操作，使得约等于O（1）可以获得父元素(自我理解)
*/

路径压缩非递归

int find_par(int x) 
{
    int r = x ;
    // 循环查找父元素
    while (s[r] != r) {
        r = s[r] ; // r 最终等于x点的父亲节点值
    }
    int i = x , j = 0; 
    while(i != r){
        j = s[i] ;
        s[i] = r ;
        i = j; 
    }
}