modify code

pull/6/head
左程云 5 years ago
parent aa598b5e46
commit 4b37110e62

@ -14,12 +14,12 @@ public class Code05_UnionFind {
} }
} }
public static class UnionSet<V> { public static class UnionFind<V> {
public HashMap<V, Node<V>> nodes; public HashMap<V, Node<V>> nodes;
public HashMap<Node<V>, Node<V>> parents; public HashMap<Node<V>, Node<V>> parents;
public HashMap<Node<V>, Integer> sizeMap; public HashMap<Node<V>, Integer> sizeMap;
public UnionSet(List<V> values) { public UnionFind(List<V> values) {
nodes = new HashMap<>(); nodes = new HashMap<>();
parents = new HashMap<>(); parents = new HashMap<>();
sizeMap = new HashMap<>(); sizeMap = new HashMap<>();
@ -38,7 +38,6 @@ public class Code05_UnionFind {
path.push(cur); path.push(cur);
cur = parents.get(cur); cur = parents.get(cur);
} }
// cur == parents.get(cur)
while (!path.isEmpty()) { while (!path.isEmpty()) {
parents.put(path.pop(), cur); parents.put(path.pop(), cur);
} }

Loading…
Cancel
Save