什么是并查集:
1、非常快的检查两个元素是否是一个集合。
2、两个元素所在各自的集合,请把他们合并在一起。
先说第一个功能:一个集合中的每个元素都会指向他的父元素,最后根元素指向他自己。判断两个元素是否是一个集合内就看他们的根元素是否是一个元素就可以了。这里有一个优化,就是一个元素可能指向他的父元素,他的父元素又指向另一个元素,另一个元素才指向根元素,在查找根元素的时候,分别把这链上的元素都指向根元素就可以了。
第二个功能:合并元素的功能,首先判断这两个集合的元素个数,把个数少的合并到个数多的上面。最后再把元素少的集合的根元素指向元素多的根元素上,就合并为一个集合了。
以下是代码:
public class UnionFind {
public static void main(String[] args) {
// TODO Auto-generated method stub
}
public static class Node {
}
public static class UnionFindSet {
//节点和他的父节点
public HashMap<Node, Node> fatherMap;
//某一个节点所在的集合一共有几个节点
public HashMap<Node, Integer> sizeMap;
//初始化
public UnionFindSet(List<Node> nodes) {
makeSets(nodes);
}
private void makeSets(List<Node> nodes) {
fatherMap = new HashMap<>();
sizeMap = new HashMap<>();
for(Node node : nodes) {
//每个node自己形成节点
fatherMap.put(node, node);
sizeMap.put(node, 1);
}
}
//找头节点同时扁平化,把所有节点指向根节点
private Node findHead(Node node) {
Node father = fatherMap.get(node);
if(father != node) {
father = findHead(father);
}
fatherMap.put(node, father);
return father;
}
//是否是一个集合
public boolean isSameSet(Node a, Node b) {
return findHead(a) == findHead(b);
}
//合并一个集合
public void union(Node a, Node b) {
if(a == null || b == null) {
return;
}
Node aHead = findHead(a);
Node bHead = findHead(b);
if(aHead != bHead) {
int aSetSize = sizeMap.get(aHead);
int bSetSize = sizeMap.get(bHead);
if(aSetSize <= bSetSize) {
fatherMap.put(aHead, bHead);
sizeMap.put(bHead, aSetSize + bSetSize);
} else {
fatherMap.put(bHead, aHead);
sizeMap.put(aHead, aSetSize + bSetSize);
}
}
}
}
}