并查集模板

int pre[MAXN];
int Find(int pos) {
    if (pre[pos] == pos) // pre是自己,代表为root节点
        return pos; 
    int r = pos; 
    while (r != pre[r]) // 找到节点pos的root
        r = pre[r];
    
    // 路径压缩
    int i = pos;
    while (i != r) { // 把沿路pos->root沿路的节点的root的pre
        //全部设置为root
        int t = pre[i];
        pre[i] = r;
        i = t;
    }
    return r;
}
void Union(int posX, int posY) { // 把两个节点的root设置为同一个
    int rootX = Find(posX), rootY = Find(posY);
    if (rootX != rootY)
        pre[rootX] = rootY;
}
最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容