数据结构与算法(Java) 40：并查集

思路 一次性给定一个Node类型的列表。初始状态下，每个Node都自成一个集合，并且父节点以及这个集合的代表节点都是其自身。并查集的功能有两个：（1）（快速）查看两个元素是否属于同一个集合；（2）将两个元素各自所在的集合合并到一起。

实现以上两个功能的方法：准备两张哈希表，parentMap和sizeMap。其中parentMap中key存放Node自身，value存放其父节点；sizeMap中key存放Node自身，value存放Node所在的集合的元素个数。对于功能（1），查看两个集合是否属于同一个集合，只需要验证这两个节点所在集合的代表节点是不是一样的就行，如果一样，那么两个节点属于同一个集合，反之不属于。在查看的同时，需要顺手做一件事情，那就是在查找当前节点的代表节点时，需要将链状结构压平，例如：A->B, B->C, C->D, D->D，查询A的代表节点时，一路向上找到D的同时，需要将结构改成：A->D, B->D, C->D, D->D。即从当前节点向上至代表节点，所有节点的父节点都变为代表节点。对于功能（2），将两个元素所在的集合合并到一起时，若两个元素来自同一个集合，那么就什么不用做，如果不是来自一个集合，那么就需要通过sizeMap比较一下两个集合的元素个数，并将小的集合的接到大的集合上，即：将小集合代表元素的父节点改为大集合的代表元素。

补充 假设样本量为N，你可以随意查询两个元素是否属于一个集合，也可以随意将两个集合合并到一起，爱调用几次调用几次，只要查询次数+合并次数整体逼近O(N)及以上，那么单次操作（不管是查询还是合并），平均时间复杂度都是O(1)，常数级别。

package algorithm.section6;

import java.util.HashMap;
import java.util.List;

public class UnionFind {
    HashMap<Node, Node> parentMap = new HashMap<>();
    HashMap<Node, Integer> sizeMap = new HashMap<>();

    public static class Node{}

    public UnionFind(List<Node> nodes){
                for (Node node : nodes){
            parentMap.put(node, node);
            sizeMap.put(node, 1);
        }
    }

    private Node findHead(Node node){
        Node parent = parentMap.get(node);
        while (parent != node)
            parent = findHead(parent);
        parentMap.put(node, parent);
        return parent;
    }

    public boolean isSameSet(Node node1, Node node2){
        return findHead(node1) == findHead(node2);
    }

    public void union(Node node1, Node node2){
        if (node1 == null || node2 == null) return;
        if (findHead(node1) != findHead(node2)){
            if (sizeMap.get(findHead(node1)) > sizeMap.get(findHead(node2))){
                parentMap.put(findHead(node2), findHead(node1));
                sizeMap.put(findHead(node1), sizeMap.get(node1) + sizeMap.get(node2));
                // 只有集合头节点的size是有效的
            } else {
                parentMap.put(findHead(node1), findHead(node2));
                sizeMap.put(findHead(node2), sizeMap.get(node1) + sizeMap.get(node2));
            }
        }
    }

    public static void main(String[] args){}
}