简介
- 二分搜索树是二叉树
- 二分搜索树的每个节点的值:大于其左子树所有节点的最大值,小于其右子树所有节点的最小值。
添加元素
根据二分搜索树的性质,新增一个元素,实现原则如下:
- 如果当前节点为空,则插入
- 如果当前不为空
- 插入元素大于当前元素:在右子树上重复,直到满足第一个条件
- 插入元素小于当前元素:在左子树上重复,直到满足第一个条件
总结:采用递归实现,从根节点开始,根据每一级的节点比较大小,最终找到空的位置插入。如果比当前节点元素小,在左子树上一直重复直到找到空位,否则在右子树上重复。
public void add(E e) {
root = add(root, e);
}
private Node add(Node node, E e) {
// 基本问题
if (node == null) {
size++;
return new Node(e);
}
// 更小问题
if (e.compareTo(node.e) < 0) {
node.left = add(node.left, e);
} else if (e.compareTo(node.e) > 0) {
node.right = add(node.right, e);
}
return node;
}
删除元素
删除比添加复杂一点,主要考虑当找到应该删除的节点时,如果这个节点本身还有孩子,那么继任者如何挑选?
考虑到二叉树的特点,可以分为如下几种基本情况:
-
只有一个孩子
继任者只能是这个孩子
-
俩孩子,那么可以从如下方案中任选其一
- 找到右子树的最小节点充当继任者(老师采用这种)
- 找到左子树的最大节点充当继任者
以2.1方案为准,最终实现如下:
- 没有孩子,则当前节点置为空
- 左子树为空
private Node remove(Node node, E e) {
if (node == null) {
return null;
}
if (e.compareTo(node.e) < 0) {
node.left = remove(node.left, e);
return node;
} else if (e.compareTo(node.e) > 0) {
node.right = remove(node.right, e);
return node;
}
// 匹配到了待删除节点,说明有节点删除,先维护size
size--;
if (node.left == null) {
return node.right;
}
if (node.right == null) {
return node.left;
}
Node successor = minimum(node.right);
successor.right = removeMin(node.right);
successor.left = node.left;
/**
* 由于removeMin中维护了size,而上面几部操作的实际结果只是将successor挪动了位置
* 因此将其size再加回来
*/
size++;
node = node.left = node.right = null; // help GC
return successor;
}
实现
扩展
问题
拍脑袋决定从哪个子树上选择继任者其实有可能造成时间上的浪费,考虑如下情况,我要删除节点17:
打眼一看,直接让16充当继任者就行了,没有必要再继续递归了。
但是由于目前没有涉及到节点高度的计算,无法判断到底哪个子树节点少(找继任者比较快)。