并查集介绍

简介

简单介绍并查集。 java 示例。

并查集

• 数据结构，在树的结构里面，每个节点会记录他的子节点。在并查集中，每个节点会记录他的父节点。
  
• 主要解决问题
  • 元素分组
• 基本操作
  • 合并，将两个不相交的集合合并成一个集合。
  • 查询，查询两个元素是否在同一个集合。

import java.util.ArrayList;
import java.util.List;


public class App {
    //存储每个元素的父节点。（因为一个元素，只有一个父节点，所以这么存储是可行的）。开始，假设父节点是自己。
    List<Integer> fatherNodes = new ArrayList<Integer>();

    public void init(int n) {
        for (int i = 0; i < n; i++) {
            fatherNodes.add(i);
        }
    }

    /**
     * 查找所属的集合。（一层层的找到父节点，父节点就是，父节点是自己的节点）。
     * 当两个节点父节点相同则，是同一个集合。
     * @param node
     * @return 根节点
     */
    public Integer find(Integer node) {
        Integer fatherNode = fatherNodes.get(node);
        if (fatherNode == node) {
            return node;
        } else {
            Integer fatherRoot = find(fatherNode);
            fatherNodes.set(node,fatherRoot); // 父节点设置为根节点。进行路径压缩。
            return fatherRoot;
        }
    }

    /**
     * 合并
     *      将 node1 的root节点。设置为 node2，的root 节点
     * 优化
     *      按照 秩 合并。深度，防止路径太长。
     * @param node1
     * @param node2
     */
    public void merger(Integer node1,Integer node2){
        fatherNodes.set(find(node1),find(node2));
    }

    public static void main(String[] args) {
        App app = new App();
        app.init(5);
        System.out.println(app.fatherNodes);
        app.merger(1,2);
        System.out.println(app.fatherNodes);
        app.merger(1,4);
        app.find(1); // 激活路径压缩
        System.out.println(app.fatherNodes);
        for (int i = 0; i < app.fatherNodes.size(); i++) {
            if (i == app.fatherNodes.get(i)){
                System.out.print(i+","); // 输出所以的父节点
            }
        }

    }
}