[关闭]
@huangyichun 2017-08-30T15:52:05.000000Z 字数 4528 阅读 930

二叉查找树

数据结构


定义:二叉查找树是一棵二叉树,且每个节点的值都大于左子树的值,小于右子树的值。

二叉查找树最坏情况下:
    查找是 N次, 插入也是N次
二叉查找树平均情况下:
    查找是 1.39lgN次, 插入也是 1.39lgN次

如下图所示:

其中数字代表二叉查找树的键,字母代表二叉查找树的值

image.png-46.7kB

我们可以通过二叉树的中序遍历二叉查找树得到一个有序的数组。

数据表示:设计一个二叉查找树类,在类中嵌套一个私有类表示二叉查找树的一个节点,每个节点都有一个键和一个值,一个左链接和一个右链接,和一个节点计数器。左链接指向一棵小于该节点的二叉排序树,右链接指向一棵大于该节点的二叉排序树。

  1. /**
  2. * 二叉查找树实现
  3. */
  4. public class BST <Key extends Comparable<Key>, Value> {
  5. private Node root; //根节点
  6. private class Node{
  7. private Key key; //键
  8. private Value val; //值
  9. private Node left, right; //左右子树的链接
  10. private int N; //以该节点为根的子树中的节点总数
  11. public Node(Key key, Value val, int N) {
  12. this.key = key;
  13. this.val = val;
  14. this.N = N;
  15. }
  16. }
  17. public void printfBST(){
  18. printfBST(root);
  19. }
  20. /**
  21. * 中序打印二叉查找树
  22. * @param node
  23. */
  24. private void printfBST(Node node){
  25. if(node == null)
  26. return;
  27. printfBST(node.left);
  28. System.out.print(node.key+" ");
  29. printfBST(node.right);
  30. }
  31. /**
  32. * 从根节点出发,查找树中排名为K的键,即正好有k个小于它的键
  33. * @param k
  34. * @return
  35. */
  36. public Key select(int k){
  37. return select(root, k).key;
  38. }
  39. /**
  40. * 删除最小键值的节点
  41. */
  42. public void deleteMin(){
  43. root = deleteMin(root);
  44. }
  45. /**
  46. * 删除键值为key的节点
  47. * @param key
  48. */
  49. public void delete(Key key){
  50. root = delete(root,key);
  51. }
  52. /**
  53. * 以Node为根节点删除某个键为key的节点
  54. * 1.首先找到值为Key的节点
  55. * 2.如果node没有右子树,返回node的左子树,
  56. * 如果node没有左子树,返回node的右子树
  57. * 3.如果node左右子树都存在,则在node的右子树中找到最小的一个
  58. * 节点,替代node
  59. * @param node
  60. * @param key
  61. * @return
  62. */
  63. private Node delete(Node node , Key key){
  64. if(node == null)
  65. return null;
  66. int cmp = key.compareTo(node.key);
  67. if(cmp < 0)
  68. node.left = delete(node.left, key);
  69. else if(cmp > 0)
  70. node.right = delete(node.right, key);
  71. else {
  72. if(node.right == null) return node.left;
  73. if(node.left == null) return node.right;
  74. Node t = node;
  75. node = min(node.right);
  76. node.right = deleteMin(t.right);
  77. node.left = t.left;
  78. }
  79. node.N = size(node.left) + size(node.right) + 1;
  80. return node;
  81. }
  82. /**
  83. * 以node为根节点删除最小的节点
  84. * @param node
  85. * @return
  86. */
  87. private Node deleteMin(Node node){
  88. if(node.left == null)
  89. return node.right;
  90. node.left = deleteMin(node.left); //删除最小节点
  91. node.N = size(node.left) + size(node.right) + 1;//更新节点数
  92. return node;
  93. }
  94. /**
  95. * 从Node节点出发,查找树中排名为K的键,即正好有k个小于它的键
  96. * @param node
  97. * @param k
  98. * @return
  99. */
  100. private Node select(Node node, int k){
  101. if(node == null)
  102. return null;
  103. int t = size(node.left);//左子树个数
  104. if(t > k) return select(node.left,k);
  105. else if(t < k) return select(node.right, k-t-1);//右子树中查找
  106. else return node;
  107. }
  108. /**
  109. * 从根节点开始查找Key
  110. * 如果存在返回,不存在找到第一个小于该key的键
  111. * @param key
  112. * @return
  113. */
  114. public Key floor(Key key){
  115. Node node = floor(root, key);
  116. if(node == null) return null;
  117. return node.key;
  118. }
  119. /**
  120. * 从node节点开始查找Key
  121. * 如果存在返回,不存在找到第一个小于该key的键
  122. * 即向下取整
  123. * @param node
  124. * @param key
  125. * @return
  126. */
  127. private Node floor(Node node, Key key){
  128. if(node == null)
  129. return null;
  130. int cmp = key.compareTo(node.key);
  131. if(cmp == 0) return node;
  132. else if(cmp < 0) return floor(node.left, key);
  133. Node t = floor(node.right, key);//查找右子树中是否有小于Key的节点
  134. if(t != null) return t;
  135. else return node;
  136. }
  137. /**
  138. * 查找根节点最大的键
  139. * @return
  140. */
  141. public Key max(){
  142. return max(root).key;
  143. }
  144. /**
  145. *查找以node节点为根节点的最大键
  146. * 如果左子树为空,则当前节点为最大值
  147. * 否则最小值就是左子树的最大值
  148. * @param node
  149. * @return
  150. */
  151. private Node max(Node node){
  152. if(node.right == null)
  153. return node;
  154. return max(node.right);
  155. }
  156. /**
  157. * 查找根节点最小的键
  158. * @return
  159. */
  160. public Key min(){
  161. return min(root).key;
  162. }
  163. /**
  164. * 查找以node节点为根节点的最小键
  165. * 如果左子树为空,则当前节点为最小值
  166. * 否则最小值就是左子树的最小值
  167. * @param node
  168. * @return
  169. */
  170. private Node min(Node node){
  171. if(node.left == null) return node;
  172. return min(node.left);
  173. }
  174. /**
  175. * 查找key存在则修改更新它的值,否则创建一个新的节点
  176. * @param key
  177. * @param value
  178. */
  179. public void put(Key key, Value value){
  180. root = put(root, key, value);
  181. }
  182. /**
  183. * 如果Key存在于以node为根节点的子树中,则更新它的值
  184. * 否则将以key和Value为键值对的新节点插入到该子树中
  185. * @param node
  186. * @param key
  187. * @param value
  188. * @return
  189. */
  190. private Node put(Node node, Key key, Value value){
  191. if(node == null)
  192. return new Node(key, value, 1);
  193. int cmp = key.compareTo(node.key);
  194. if(cmp < 0) node.left = put(node.left, key, value);
  195. else if(cmp > 0) node.right = put(node.right, key, value);
  196. else node.val = value;
  197. node.N = size(node.left) + size(node.right) + 1;//更新节点数
  198. return node;
  199. }
  200. /**
  201. * 获取Key对应的Value
  202. * @param key
  203. * @return
  204. */
  205. public Value get(Key key){
  206. return get(root, key);
  207. }
  208. /**
  209. * 在以node为根节点的子树中查找并返回Key所对应的值
  210. * @param node
  211. * @param key
  212. * @return
  213. */
  214. private Value get(Node node, Key key){
  215. if(node == null)
  216. return null;
  217. int cmp = key.compareTo(node.key);
  218. if(cmp < 0) return get(node.left, key);
  219. else if(cmp > 0) return get(node.right, key);
  220. else return node.val;
  221. }
  222. /**
  223. * 获取二叉查找树节点个数
  224. * 满足 size(x) = size(x.left) + size(x.right) + 1;
  225. * 对于空链接size(x)返回0
  226. * @return 返回节点个数
  227. */
  228. public int size(){
  229. return size(root);
  230. }
  231. private int size(Node node){
  232. if(node == null)
  233. return 0;
  234. else
  235. return node.N;
  236. }
  237. }

构建上图的二叉查找树,对于里面的方法进行测试:

  1. /**
  2. * Created by huangyichun on 2017/5/1.
  3. */
  4. public class Test {
  5. public static void main(String[] args) {
  6. BST<Integer,Character> bst = new BST<Integer, Character>();
  7. bst.put(15,'A');
  8. bst.put(13,'B');
  9. bst.put(11,'D');
  10. bst.put(8,'H');
  11. bst.put(14,'E');
  12. bst.put(20,'C');
  13. bst.put(17,'F');
  14. bst.put(11,'G');
  15. bst.printfBST();
  16. System.out.println();
  17. System.out.println("最大键:" + bst.max());//打印最大键
  18. System.out.println("最小键:" + bst.min());//打印最小键
  19. System.out.println("Key= 15, value=" + bst.get(15));
  20. System.out.println("节点个数为:" + bst.size());
  21. bst.delete(13);//删除13这个节点
  22. bst.printfBST();
  23. // bst.deleteMin();
  24. // bst.printfBST();
  25. }
  26. }

输出结果为:

8 11 13 14 15 17 20 
最大键:20
最小键:8
Key= 15, value=A
节点个数为:7
8 11 14 15 17 20 
添加新批注
在作者公开此批注前,只有你和作者可见。
回复批注