1 public class BinaryTree { 2 3 // 二叉树的根节点 4 public TreeNode rootNode ; 5 // 记录搜索深度 6 public int count; 7 8 /** 9 * 利用传入一个数组来建立二叉树 10 */ 11 public BinaryTree(int[] data) { 12 for (int i = 0; i < data. length; i++) { 13 addNodeToTree(data[i]); 14 } 15 } 16 17 /** 18 * 将指定的值加入到二叉树中适当的节点 19 */ 20 private void addNodeToTree(int value) { 21 TreeNode currentNode = rootNode; 22 // 建立树根 23 if (rootNode == null) { 24 rootNode = new TreeNode(value); 25 return; 26 } 27 28 // 建立二叉树 29 while (true) { 30 // 新增的value比节点的value小,则在左子树 31 if (value < currentNode.value ) { 32 if (currentNode.leftNode == null) { 33 currentNode. leftNode = new TreeNode(value); 34 return; 35 } else { 36 currentNode = currentNode. leftNode; 37 } 38 } else { // 新增的value比节点的value大,在右子树 39 if (currentNode.rightNode == null) { 40 currentNode. rightNode = new TreeNode(value); 41 return; 42 } else { 43 currentNode = currentNode. rightNode; 44 } 45 } 46 } 47 } 48 49 /** 50 * 中序遍历(左子树 -树根- 右子树) 51 */ 52 public void inOrder(TreeNode node) { 53 if (node != null) { 54 inOrder(node. leftNode); 55 System. out.print("[" + node.value + "]"); 56 inOrder(node. rightNode); 57 } 58 } 59 60 /** 61 * 前序遍历(树根 -左子树- 右子树) 62 */ 63 public void preOrder(TreeNode node) { 64 if (node != null) { 65 System. out.print("[" + node.value + "]"); 66 preOrder(node. leftNode); 67 preOrder(node. rightNode); 68 } 69 } 70 71 /** 72 * 后序遍历(左子树 -右子树- 树根) 73 */ 74 public void postOrder(TreeNode node) { 75 if (node != null) { 76 postOrder(node. leftNode); 77 postOrder(node. rightNode); 78 System. out.print("[" + node.value + "]"); 79 } 80 } 81 82 /** 83 * 从二叉树中查找指定value 84 */ 85 public boolean findTree(TreeNode node, int value) { 86 if (node == null) { 87 System. out.println("共搜索" + count + "次"); 88 return false; 89 } else if (node.value == value) { 90 System. out.println("共搜索" + count + "次"); 91 return true; 92 } else if (value < node.value) { 93 count++; 94 return findTree(node.leftNode , value); 95 } else { 96 count++; 97 return findTree(node.rightNode , value); 98 } 99 } 100 101 /** 102 * 利用中序遍历进行排序 103 */ 104 public void sort() { 105 this.inOrder(rootNode ); 106 } 107 108 class TreeNode { 109 int value ; 110 TreeNode leftNode; 111 TreeNode rightNode; 112 113 public TreeNode(int value) { 114 this.value = value; 115 this.leftNode = null; 116 this.rightNode = null; 117 } 118 } 119 120 public static void main(String[] args) { 121 int[] content = { 50, 35, 27, 45, 40, 48, 78, 56, 90 }; 122 123 BinaryTree tree = new BinaryTree(content); 124 System. out.println("前序遍历:" ); 125 tree.preOrder(tree. rootNode); 126 System. out.println("\n中序遍历:" ); 127 tree.inOrder(tree. rootNode); 128 System. out.println("\n后序遍历:" ); 129 tree.postOrder(tree. rootNode); 130 131 System. out.println("\n\n开始搜索:" ); 132 boolean isFind = tree.findTree(tree.rootNode, 48); 133 System. out.println("是否搜索到" + 48 + ":" + isFind); 134 135 System. out.println("\n进行排序:" ); 136 tree.sort(); 137 } 138 }
1 前序遍历: 2 [50][35][27][45][40][48][78][56][90] 3 中序遍历: 4 [27][35][40][45][48][50][56][78][90] 5 后序遍历: 6 [27][40][48][45][35][56][90][78][50] 7 8 开始搜索: 9 共搜索3次 10 是否搜索到48:true 11 12 进行排序: 13 [27][35][40][45][48][50][56][78][90]
1 public class TreeMap<K,V> 2 extends AbstractMap<K,V> 3 implements NavigableMap<K,V>, Cloneable, java.io.Serializable
1 // 比较器 2 private final Comparator<? super K> comparator; 3 4 // 红黑树根节点 5 private transient Entry<K,V> root = null; 6 7 // 集合元素数量 8 private transient int size = 0; 9 10 // "fail-fast"集合修改记录 11 private transient int modCount = 0;
这里的Comparator是一个比较器,这里不详细讲解,后面会单独进行分析,这里只要明白,一个类实现了Comparator接口并重写其compare方法,就能进行比较大小。Entry是树的节点类,我们来看一下Entry的定义:
1 static final class Entry<K,V> implements Map.Entry<K,V> { 2 K key; 3 V value; 4 // 左孩子节点 5 Entry<K,V> left = null; 6 // 右孩子节点 7 Entry<K,V> right = null; 8 // 父节点 9 Entry<K,V> parent; 10 // 红黑树用来表示节点颜色的属性,默认为黑色 11 boolean color = BLACK; 12 13 /** 14 * 用key,value和父节点构造一个Entry,默认为黑色 15 */ 16 Entry(K key, V value, Entry<K,V> parent) { 17 this.key = key; 18 this.value = value; 19 this.parent = parent; 20 } 21 22 public K getKey() { 23 return key ; 24 } 25 26 public V getValue() { 27 return value ; 28 } 29 30 public V setValue(V value) { 31 V oldValue = this.value ; 32 this.value = value; 33 return oldValue; 34 } 35 36 public boolean equals(Object o) { 37 if (!(o instanceof Map.Entry)) 38 return false; 39 Map.Entry<?,?> e = (Map.Entry<?,?>)o; 40 41 return valEquals( key,e.getKey()) && valEquals( value,e.getValue()); 42 } 43 44 public int hashCode() { 45 int keyHash = (key ==null ? 0 : key.hashCode()); 46 int valueHash = (value ==null ? 0 : value.hashCode()); 47 return keyHash ^ valueHash; 48 } 49 50 public String toString() { 51 return key + "=" + value; 52 } 53 }
1 /** 2 * 默认构造方法,comparator为空,代表使用key的自然顺序来维持TreeMap的顺序,这里要求key必须实现Comparable接口 3 */ 4 public TreeMap() { 5 comparator = null; 6 } 7 8 /** 9 * 用指定的比较器构造一个TreeMap 10 */ 11 public TreeMap(Comparator<? super K> comparator) { 12 this.comparator = comparator; 13 } 14 15 /** 16 * 构造一个指定map的TreeMap,同样比较器comparator为空,使用key的自然顺序排序 17 */ 18 public TreeMap(Map<? extends K, ? extends V> m) { 19 comparator = null; 20 putAll(m); 21 } 22 23 /** 24 * 构造一个指定SortedMap的TreeMap,根据SortedMap的比较器来来维持TreeMap的顺序 25 */ 26 public TreeMap(SortedMap<K, ? extends V> m) { 27 comparator = m.comparator(); 28 try { 29 buildFromSorted(m.size(), m.entrySet().iterator(), null, null); 30 } catch (java.io.IOException cannotHappen) { 31 } catch (ClassNotFoundException cannotHappen) { 32 } 33 }
1 public V put(K key, V value) { 2 // 根节点 3 Entry<K,V> t = root; 4 // 如果根节点为空,则直接创建一个根节点,返回 5 if (t == null) { 6 // TBD: 7 // 5045147: (coll) Adding null to an empty TreeSet should 8 // throw NullPointerException 9 // 10 // compare(key, key); // type check 11 root = new Entry<K,V>(key, value, null); 12 size = 1; 13 modCount++; 14 return null; 15 } 16 // 记录比较结果 17 int cmp; 18 Entry<K,V> parent; 19 // split comparator and comparable paths 20 // 当前使用的比较器 21 Comparator<? super K> cpr = comparator ; 22 // 如果比较器不为空,就是用指定的比较器来维护TreeMap的元素顺序 23 if (cpr != null) { 24 // do while循环,查找key要插入的位置(也就是新节点的父节点是谁) 25 do { 26 // 记录上次循环的节点t 27 parent = t; 28 // 比较当前节点的key和新插入的key的大小 29 cmp = cpr.compare(key, t. key); 30 // 新插入的key小的话,则以当前节点的左孩子节点为新的比较节点 31 if (cmp < 0) 32 t = t. left; 33 // 新插入的key大的话,则以当前节点的右孩子节点为新的比较节点 34 else if (cmp > 0) 35 t = t. right; 36 else 37 // 如果当前节点的key和新插入的key想的的话,则覆盖map的value,返回 38 return t.setValue(value); 39 // 只有当t为null,也就是没有要比较节点的时候,代表已经找到新节点要插入的位置 40 } while (t != null); 41 } 42 else { 43 // 如果比较器为空,则使用key作为比较器进行比较 44 // 这里要求key不能为空,并且必须实现Comparable接口 45 if (key == null) 46 throw new NullPointerException(); 47 Comparable<? super K> k = (Comparable<? super K>) key; 48 // 和上面一样,喜欢查找新节点要插入的位置 49 do { 50 parent = t; 51 cmp = k.compareTo(t. key); 52 if (cmp < 0) 53 t = t. left; 54 else if (cmp > 0) 55 t = t. right; 56 else 57 return t.setValue(value); 58 } while (t != null); 59 } 60 // 找到新节点的父节点后,创建节点对象 61 Entry<K,V> e = new Entry<K,V>(key, value, parent); 62 // 如果新节点key的值小于父节点key的值,则插在父节点的左侧 63 if (cmp < 0) 64 parent. left = e; 65 // 如果新节点key的值大于父节点key的值,则插在父节点的右侧 66 else 67 parent. right = e; 68 // 插入新的节点后,为了保持红黑树平衡,对红黑树进行调整 69 fixAfterInsertion(e); 70 // map元素个数+1 71 size++; 72 modCount++; 73 return null; 74 } 75 76 77 /** 新增节点后对红黑树的调整方法 */ 78 private void fixAfterInsertion(Entry<K,V> x) { 79 // 将新插入节点的颜色设置为红色 80 x. color = RED; 81 82 // while循环,保证新插入节点x不是根节点或者新插入节点x的父节点不是红色(这两种情况不需要调整) 83 while (x != null && x != root && x. parent.color == RED) { 84 // 如果新插入节点x的父节点是祖父节点的左孩子 85 if (parentOf(x) == leftOf(parentOf (parentOf(x)))) { 86 // 取得新插入节点x的叔叔节点 87 Entry<K,V> y = rightOf(parentOf (parentOf(x))); 88 // 如果新插入x的父节点是红色-------------------① 89 if (colorOf(y) == RED) { 90 // 将x的父节点设置为黑色 91 setColor(parentOf (x), BLACK); 92 // 将x的叔叔节点设置为黑色 93 setColor(y, BLACK); 94 // 将x的祖父节点设置为红色 95 setColor(parentOf (parentOf(x)), RED); 96 // 将x指向祖父节点,如果x的祖父节点的父节点是红色,按照上面的步奏继续循环 97 x = parentOf(parentOf (x)); 98 } else { 99 // 如果新插入x的叔叔节点是黑色或缺少,且x的父节点是祖父节点的右孩子-------------------② 100 if (x == rightOf( parentOf(x))) { 101 // 左旋父节点 102 x = parentOf(x); 103 rotateLeft(x); 104 } 105 // 如果新插入x的叔叔节点是黑色或缺少,且x的父节点是祖父节点的左孩子-------------------③ 106 // 将x的父节点设置为黑色 107 setColor(parentOf (x), BLACK); 108 // 将x的祖父节点设置为红色 109 setColor(parentOf (parentOf(x)), RED); 110 // 右旋x的祖父节点 111 rotateRight( parentOf(parentOf (x))); 112 } 113 } else { // 如果新插入节点x的父节点是祖父节点的右孩子,下面的步奏和上面的相似,只不过左旋右旋的区分,不在细讲 114 Entry<K,V> y = leftOf(parentOf (parentOf(x))); 115 if (colorOf(y) == RED) { 116 setColor(parentOf (x), BLACK); 117 setColor(y, BLACK); 118 setColor(parentOf (parentOf(x)), RED); 119 x = parentOf(parentOf (x)); 120 } else { 121 if (x == leftOf( parentOf(x))) { 122 x = parentOf(x); 123 rotateRight(x); 124 } 125 setColor(parentOf (x), BLACK); 126 setColor(parentOf (parentOf(x)), RED); 127 rotateLeft( parentOf(parentOf (x))); 128 } 129 } 130 } 131 // 最后将根节点设置为黑色,不管当前是不是红色,反正根节点必须是黑色 132 root.color = BLACK; 133 } 134 135 /** 136 * 对红黑树的节点(x)进行左旋转 137 * 138 * 左旋示意图(对节点x进行左旋): 139 * px px 140 * / / 141 * x y 142 * / \ --(左旋)-- / \ 143 * lx y x ry 144 * / \ / \ 145 * ly ry lx ly 146 * 147 */ 148 private void rotateLeft(Entry<K,V> p) { 149 if (p != null) { 150 // 取得要选择节点p的右孩子 151 Entry<K,V> r = p. right; 152 // "p"和"r的左孩子"的相互指向... 153 // 将"r的左孩子"设为"p的右孩子" 154 p. right = r.left ; 155 // 如果r的左孩子非空,将"p"设为"r的左孩子的父亲" 156 if (r.left != null) 157 r. left.parent = p; 158 159 // "p的父亲"和"r"的相互指向... 160 // 将"p的父亲"设为"y的父亲" 161 r. parent = p.parent ; 162 // 如果"p的父亲"是空节点,则将r设为根节点 163 if (p.parent == null) 164 root = r; 165 // 如果p是它父节点的左孩子,则将r设为"p的父节点的左孩子" 166 else if (p.parent. left == p) 167 p. parent.left = r; 168 else 169 // 如果p是它父节点的左孩子,则将r设为"p的父节点的左孩子" 170 p. parent.right = r; 171 // "p"和"r"的相互指向... 172 // 将"p"设为"r的左孩子" 173 r. left = p; 174 // 将"p的父节点"设为"r" 175 p. parent = r; 176 } 177 } 178 179 180 /** 181 * 对红黑树的节点进行右旋转 182 * 183 * 右旋示意图(对节点y进行右旋): 184 * py py 185 * / / 186 * y x 187 * / \ --(右旋)-- / \ 188 * x ry lx y 189 * / \ / \ 190 * lx rx rx ry 191 * 192 */ 193 private void rotateRight(Entry<K,V> p) { 194 if (p != null) { 195 // 取得要选择节点p的左孩子 196 Entry<K,V> l = p. left; 197 // 将"l的右孩子"设为"p的左孩子" 198 p. left = l.right ; 199 // 如果"l的右孩子"不为空的话,将"p"设为"l的右孩子的父亲" 200 if (l.right != null) l. right.parent = p; 201 // 将"p的父亲"设为"l的父亲" 202 l. parent = p.parent ; 203 // 如果"p的父亲"是空节点,则将l设为根节点 204 if (p.parent == null) 205 root = l; 206 // 如果p是它父节点的右孩子,则将l设为"p的父节点的右孩子" 207 else if (p.parent. right == p) 208 p. parent.right = l; 209 //如果p是它父节点的左孩子,将l设为"p的父节点的左孩子" 210 else p.parent .left = l; 211 // 将"p"设为"l的右孩子" 212 l. right = p; 213 // 将"l"设为"p父节点" 214 p. parent = l; 215 } 216 }
1 public V remove(Object key) { 2 // 根据key查找到对应的节点对象 3 Entry<K,V> p = getEntry(key); 4 if (p == null) 5 return null; 6 7 // 记录key对应的value,供返回使用 8 V oldValue = p. value; 9 // 删除节点 10 deleteEntry(p); 11 return oldValue; 12 } 13 14 15 private void deleteEntry(Entry<K,V> p) { 16 modCount++; 17 // map容器的元素个数减一 18 size--; 19 20 // If strictly internal, copy successor's element to p and then make p 21 // point to successor. 22 // 如果被删除的节点p的左孩子和右孩子都不为空,则查找其替代节点-----------这里表示要删除的节点有两个孩子(3) 23 if (p.left != null && p. right != null) { 24 // 查找p的替代节点 25 Entry<K,V> s = successor (p); 26 p. key = s.key ; 27 p. value = s.value ; 28 // 将p指向替代节点,※※※※※※从此之后的p不再是原先要删除的节点p,而是替代者p(就是图解里面讲到的M) ※※※※※※ 29 p = s; 30 } // p has 2 children 31 32 // Start fixup at replacement node, if it exists. 33 // replacement为替代节点p的继承者(就是图解里面讲到的N),p的左孩子存在则用p的左孩子替代,否则用p的右孩子 34 Entry<K,V> replacement = (p. left != null ? p.left : p. right); 35 36 if (replacement != null) { // 如果上面的if有两个孩子不通过--------------这里表示要删除的节点只有一个孩子(2) 37 // Link replacement to parent 38 // 将p的父节点拷贝给替代节点 39 replacement. parent = p.parent ; 40 // 如果替代节点p的父节点为空,也就是p为跟节点,则将replacement设置为根节点 41 if (p.parent == null) 42 root = replacement; 43 // 如果替代节点p是其父节点的左孩子,则将replacement设置为其父节点的左孩子 44 else if (p == p.parent. left) 45 p. parent.left = replacement; 46 // 如果替代节点p是其父节点的左孩子,则将replacement设置为其父节点的右孩子 47 else 48 p. parent.right = replacement; 49 50 // Null out links so they are OK to use by fixAfterDeletion. 51 // 将替代节点p的left、right、parent的指针都指向空,即解除前后引用关系(相当于将p从树种摘除),使得gc可以回收 52 p. left = p.right = p.parent = null; 53 54 // Fix replacement 55 // 如果替代节点p的颜色是黑色,则需要调整红黑树以保持其平衡 56 if (p.color == BLACK) 57 fixAfterDeletion(replacement); 58 } else if (p.parent == null) { // return if we are the only node. 59 // 如果要替代节点p没有父节点,代表p为根节点,直接删除即可 60 root = null; 61 } else { // No children. Use self as phantom replacement and unlink. 62 // 判断进入这里说明替代节点p没有孩子--------------这里表示没有孩子则直接删除(1) 63 // 如果p的颜色是黑色,则调整红黑树 64 if (p.color == BLACK) 65 fixAfterDeletion(p); 66 // 下面删除替代节点p 67 if (p.parent != null) { 68 // 解除p的父节点对p的引用 69 if (p == p.parent .left) 70 p. parent.left = null; 71 else if (p == p.parent. right) 72 p. parent.right = null; 73 // 解除p对p父节点的引用 74 p. parent = null; 75 } 76 } 77 } 78 79 /** 80 * 查找要删除节点的替代节点 81 */ 82 static <K,V> TreeMap.Entry<K,V> successor(Entry<K,V> t) { 83 if (t == null) 84 return null; 85 // 查找右子树的最左孩子 86 else if (t.right != null) { 87 Entry<K,V> p = t. right; 88 while (p.left != null) 89 p = p. left; 90 return p; 91 } else { // 查找左子树的最右孩子 92 Entry<K,V> p = t. parent; 93 Entry<K,V> ch = t; 94 while (p != null && ch == p. right) { 95 ch = p; 96 p = p. parent; 97 } 98 return p; 99 } 100 } 101 102 /** From CLR */ 103 private void fixAfterDeletion(Entry<K,V> x) { 104 // while循环,保证要删除节点x不是跟节点,并且是黑色(根节点和红色不需要调整) 105 while (x != root && colorOf (x) == BLACK) { 106 // 如果要删除节点x是其父亲的左孩子 107 if (x == leftOf( parentOf(x))) { 108 // 取出要删除节点x的兄弟节点 109 Entry<K,V> sib = rightOf(parentOf (x)); 110 111 // 如果删除节点x的兄弟节点是红色---------------------------① 112 if (colorOf(sib) == RED) { 113 // 将x的兄弟节点颜色设置为黑色 114 setColor(sib, BLACK); 115 // 将x的父节点颜色设置为红色 116 setColor(parentOf (x), RED); 117 // 左旋x的父节点 118 rotateLeft( parentOf(x)); 119 // 将sib重新指向旋转后x的兄弟节点 ,进入else的步奏③ 120 sib = rightOf(parentOf (x)); 121 } 122 123 // 如果x的兄弟节点的两个孩子都是黑色-------------------------③ 124 if (colorOf(leftOf(sib)) == BLACK && 125 colorOf(rightOf (sib)) == BLACK) { 126 // 将兄弟节点的颜色设置为红色 127 setColor(sib, RED); 128 // 将x的父节点指向x,如果x的父节点是黑色,需要将x的父节点整天看做一个节点继续调整-------------------------② 129 x = parentOf(x); 130 } else { 131 // 如果x的兄弟节点右孩子是黑色,左孩子是红色-------------------------④ 132 if (colorOf(rightOf(sib)) == BLACK) { 133 // 将x的兄弟节点的左孩子设置为黑色 134 setColor(leftOf (sib), BLACK); 135 // 将x的兄弟节点设置为红色 136 setColor(sib, RED); 137 // 右旋x的兄弟节点 138 rotateRight(sib); 139 // 将sib重新指向旋转后x的兄弟节点,进入步奏⑤ 140 sib = rightOf(parentOf (x)); 141 } 142 // 如果x的兄弟节点右孩子是红色-------------------------⑤ 143 setColor(sib, colorOf (parentOf(x))); 144 // 将x的父节点设置为黑色 145 setColor(parentOf (x), BLACK); 146 // 将x的兄弟节点的右孩子设置为黑色 147 setColor(rightOf (sib), BLACK); 148 // 左旋x的父节点 149 rotateLeft( parentOf(x)); 150 // 达到平衡,将x指向root,退出循环 151 x = root; 152 } 153 } else { // symmetric // 如果要删除节点x是其父亲的右孩子,和上面情况一样,这里不再细讲 154 Entry<K,V> sib = leftOf(parentOf (x)); 155 156 if (colorOf(sib) == RED) { 157 setColor(sib, BLACK); 158 setColor(parentOf (x), RED); 159 rotateRight( parentOf(x)); 160 sib = leftOf(parentOf (x)); 161 } 162 163 if (colorOf(rightOf(sib)) == BLACK && 164 colorOf(leftOf (sib)) == BLACK) { 165 setColor(sib, RED); 166 x = parentOf(x); 167 } else { 168 if (colorOf(leftOf(sib)) == BLACK) { 169 setColor(rightOf (sib), BLACK); 170 setColor(sib, RED); 171 rotateLeft(sib); 172 sib = leftOf(parentOf (x)); 173 } 174 setColor(sib, colorOf (parentOf(x))); 175 setColor(parentOf (x), BLACK); 176 setColor(leftOf (sib), BLACK); 177 rotateRight( parentOf(x)); 178 x = root; 179 } 180 } 181 } 182 183 setColor(x, BLACK); 184 }
1 public V get(Object key) { 2 Entry<K,V> p = getEntry(key); 3 return (p==null ? null : p. value); 4 } 5 6 final Entry<K,V> getEntry(Object key) { 7 // Offload comparator-based version for sake of performance 8 if (comparator != null) 9 // 如果比较器为空,只是用key作为比较器查询 10 return getEntryUsingComparator(key); 11 if (key == null) 12 throw new NullPointerException(); 13 Comparable<? super K> k = (Comparable<? super K>) key; 14 // 取得root节点 15 Entry<K,V> p = root; 16 // 从root节点开始查找,根据比较器判断是在左子树还是右子树 17 while (p != null) { 18 int cmp = k.compareTo(p.key ); 19 if (cmp < 0) 20 p = p. left; 21 else if (cmp > 0) 22 p = p. right; 23 else 24 return p; 25 } 26 return null; 27 } 28 29 final Entry<K,V> getEntryUsingComparator(Object key) { 30 K k = (K) key; 31 Comparator<? super K> cpr = comparator ; 32 if (cpr != null) { 33 Entry<K,V> p = root; 34 while (p != null) { 35 int cmp = cpr.compare(k, p.key ); 36 if (cmp < 0) 37 p = p. left; 38 else if (cmp > 0) 39 p = p. right; 40 else 41 return p; 42 } 43 } 44 return null; 45 }