数据结构-二分搜索树

简介

1. 二分搜索树是二叉树
2. 二分搜索树的每个节点的值：大于其左子树所有节点的最大值，小于其右子树所有节点的最小值。

添加元素

根据二分搜索树的性质，新增一个元素，实现原则如下：

1. 如果当前节点为空，则插入
2. 如果当前不为空
   1. 插入元素大于当前元素：在右子树上重复，直到满足第一个条件
   2. 插入元素小于当前元素：在左子树上重复，直到满足第一个条件

总结：采用递归实现，从根节点开始，根据每一级的节点比较大小，最终找到空的位置插入。如果比当前节点元素小，在左子树上一直重复直到找到空位，否则在右子树上重复。

    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;
    }

删除元素

删除比添加复杂一点，主要考虑当找到应该删除的节点时，如果这个节点本身还有孩子，那么继任者如何挑选？

考虑到二叉树的特点，可以分为如下几种基本情况：

1. 只有一个孩子

   继任者只能是这个孩子

2. 俩孩子，那么可以从如下方案中任选其一

   1. 找到右子树的最小节点充当继任者（老师采用这种）
   2. 找到左子树的最大节点充当继任者

以2.1方案为准，最终实现如下：

1. 没有孩子，则当前节点置为空
2. 左子树为空

    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;
    }

实现

我的BST源码

扩展

• 集合BSTSet

• 映射BSTMap

问题

拍脑袋决定从哪个子树上选择继任者其实有可能造成时间上的浪费，考虑如下情况，我要删除节点17：

打眼一看，直接让16充当继任者就行了，没有必要再继续递归了。

但是由于目前没有涉及到节点高度的计算，无法判断到底哪个子树节点少（找继任者比较快）。