用Java实现二叉查找树

回眸只為那壹抹淺笑 提交于 2019-11-30 19:46:13

二叉查找树的实现

1. 原理

  二叉查找树,又称为二叉排序树、二叉搜索树。对于树中每一个节点X,它的左子树中所有项的值小于X中的项,而它的右子树中所有项的值大于X中的项。二叉查找树的平均深度为O(log N),搜索元素的时间复杂度也是O(log N)。是两种库集合类TreeSet、TreeMap实现的基础。

2. public API

void makeEmpty( )      --> 置空boolean isEmpty( )     --> 判空AnyType findMin( )     --> 寻找最小值AnyType findMax( )     --> 寻找最大值boolean contains( x )  --> 是否存在元素xvoid insert( x )       --> 插入元素xvoid remove( x )       --> 删除元素xvoid printTree( )      --> 遍历二叉树

3. 核心思想图解:递归

!寻找最小值

此处用递归实现:

!寻找最大值

此处用非递归实现,也可以用递归实现:

!是否存在元素x

从root开始往下找,找到含有项X的节点,则此操作返回true,没有找到则返回false。

!插入元素x

从root开始往下找到合适的插入位置,然后插入。

!删除元素x

从root开始往下找到元素x,找到则删除,并且处理好后续工作。

4. BinarySearchTree代码实现

类中,大量使用方法来调用递归方法的技巧,很好地体现了面向对象的封装性。
  1 /**
  2  * @author: wenhx
  3  * @date: Created in 2019/10/8 19:41 (之前)
  4  * @description: 二叉查找树的实现
  5  */
  6 public class BinarySearchTree<AnyType extends Comparable<? super AnyType>> {
  7 ​
  8     /**
  9      * 树的根节点
 10      */
 11     private BinaryNode<AnyType> root;
 12 ​
 13     /**
 14      * 定义树的节点(内部类)
 15      */
 16     private static class BinaryNode<AnyType> {
 17 ​
 18         AnyType element;            // 元素值
 19         BinaryNode<AnyType> left;   // 左孩子
 20         BinaryNode<AnyType> right;  // 右孩子
 21 ​
 22         // 节点的构造器:初始化一个树的节点
 23         BinaryNode(AnyType theElement) {
 24             this(theElement, null, null);
 25         }
 26 ​
 27         BinaryNode(AnyType theElement, BinaryNode<AnyType> lt, BinaryNode<AnyType> rt) {
 28             element = theElement;
 29             left = lt;
 30             right = rt;
 31         }
 32     }
 33 ​
 34     /**
 35      * 二叉排序树的构造器:初始化根节点
 36      */
 37     public BinarySearchTree() {
 38         root = null;
 39     }
 40 ​
 41     /**
 42      * 置空
 43      */
 44     public void makeEmpty() {
 45         root = null;
 46     }
 47 ​
 48     /**
 49      * 判空
 50      */
 51     public boolean isEmpty() {
 52         return root == null;
 53     }
 54 ​
 55     /**
 56      * 寻找最小值
 57      */
 58     public AnyType findMin() {
 59         if (isEmpty()) {
 60             throw new RuntimeException();
 61         }
 62         return findMin(root).element;
 63     }
 64 ​
 65     /**
 66      * 寻找最大值
 67      */
 68     public AnyType findMax() {
 69         if (isEmpty()) {
 70             throw new RuntimeException();
 71         }
 72         return findMax(root).element;
 73     }
 74 ​
 75 ​
 76     /**
 77      * 是否存在元素x
 78      */
 79     public boolean contains(AnyType x) {
 80         return contains(x, root);
 81     }
 82 ​
 83     /**
 84      * 插入元素x
 85      */
 86     public void insert(AnyType x) {
 87         root = insert(x, root);
 88     }
 89 ​
 90     /**
 91      * 删除元素x
 92      */
 93     public void remove(AnyType x) {
 94         root = remove(x, root);
 95     }
 96 ​
 97     /**
 98      * 遍历此二叉树
 99      */
100     public void printTree() {
101         if (isEmpty()) {
102             System.out.println("Empty tree");
103         } else {
104             printTree(root);
105         }
106     }
107 ​
108     /**
109      * 寻找最小值(内部方法):此处用递归实现
110      */
111     private BinaryNode<AnyType> findMin(BinaryNode<AnyType> t) {
112         if (t == null) {
113             return null;
114         } else if (t.left == null) {
115             return t;
116         }
117         return findMin(t.left);
118     }
119 ​
120     /**
121      * 寻找最大值(内部方法):此处用非递归实现
122      */
123     private BinaryNode<AnyType> findMax(BinaryNode<AnyType> t) {
124         if (t != null) {
125             while (t.right != null) {
126                 t = t.right;
127             }
128         }
129         return t;
130     }
131 ​
132     /**
133      * 是否存在元素x(内部方法)
134      */
135     private boolean contains(AnyType x, BinaryNode<AnyType> t) {
136         /**
137          * 跳出递归的条件
138          */
139         if (t == null) {
140             return false;
141         }
142 ​
143         /**
144          * 如果x小于节点值,则递归到左孩子;
145          * 如果x大于节点值,则递归到右孩子;
146          * 如果x等于节点值,则找到。
147          */
148         int compareResult = x.compareTo(t.element);
149 ​
150         if (compareResult < 0) {
151             return contains(x, t.left);
152         } else if (compareResult > 0) {
153             return contains(x, t.right);
154         } else {
155             return true;
156         }
157 ​
158     }
159 ​
160     /**
161      * 插入元素x(内部方法)
162      */
163     private BinaryNode<AnyType> insert(AnyType x, BinaryNode<AnyType> t) {
164         /**
165          * 跳出递归的条件
166          */
167         if (t == null) {
168             return new BinaryNode<>(x, null, null);
169         }
170 ​
171         /**
172          * 如果x小于节点值,则递归到左孩子;
173          * 如果x大于节点值,则递归到右孩子;
174          * 如果x等于节点值,则说明已有元素x,无需操作。
175          */
176         int compareResult = x.compareTo(t.element);
177 ​
178         if (compareResult < 0) {
179             t.left = insert(x, t.left);
180         } else if (compareResult > 0) {
181             t.right = insert(x, t.right);
182         } else {
183         }
184         return t;
185 ​
186     }
187 ​
188     /**
189      * 删除元素x(内部方法)
190      */
191     private BinaryNode<AnyType> remove(AnyType x, BinaryNode<AnyType> t) {
192         /**
193          * 跳出递归的条件
194          */
195         if (t == null) {
196             return t;   // Item not found; do nothing
197         }
198 ​
199         /**
200          * 如果x小于节点值,则递归到左孩子;
201          * 如果x大于节点值,则递归到右孩子;
202          * 如果x等于节点值,则要删除此节点。
203          */
204         int compareResult = x.compareTo(t.element);
205 ​
206         if (compareResult < 0) {
207             t.left = remove(x, t.left);
208         } else if (compareResult > 0) {
209             t.right = remove(x, t.right);
210         } else if (t.left != null && t.right != null) {
211             // 要删除的节点有两个孩子(可选用右孩子最小元素/左孩子最大元素上调)
212             t.element = findMin(t.right).element;
213             t.right = remove(t.element, t.right);
214         } else {
215             // 要删除的节点有一个孩子或者没有孩子
216             t = (t.left != null) ? t.left : t.right;
217         }
218         return t;
219     }
220 ​
221     /**
222      * 遍历此二叉树(内部方法)
223      */
224     private void printTree(BinaryNode<AnyType> t) {
225         // 中序遍历-->即递增顺序
226         if (t != null) {
227             printTree(t.left);
228             System.out.println(t.element);
229             printTree(t.right);
230         }
231     }
232 ​
233     /**
234      * 求树的深度(内部方法)
235      */
236     private int height(BinaryNode<AnyType> t) {
237         if (t == null) {
238             return -1;
239         } else {
240             return 1 + Math.max(height(t.left), height(t.right));
241         }
242     }
243 ​
244     /**
245      * 主方法用来测试
246      */
247     public static void main(String[] args) {
248         BinarySearchTree<Integer> t = new BinarySearchTree<>();
249         t.insert(6);
250         t.insert(3);
251         t.insert(9);
252         t.insert(2);
253         t.insert(5);
254         t.insert(8);
255         t.insert(10);
256         t.printTree();
257         t.insert(4);
258     }
259 }

 

okay,今天就到这啦,一定要掌握这种数据结构哈,真的很重要!!!

易学教程内所有资源均来自网络或用户发布的内容,如有违反法律规定的内容欢迎反馈
该文章没有解决你所遇到的问题?点击提问,说说你的问题,让更多的人一起探讨吧!