算法小抄学习笔记 — 6.二叉搜索树（二）

1 二叉搜索树框架

TreeNode traverse(TreeNode root, int x) {
    if (root == null) return null;
    if (root.val == x) {
        return root;
    }
    if (root.val > x) {
        return traverse(root.right, x);
    }
    if (root.val < x) {
        return traverse(root.right, x);
    }
}

虽然上述只是BST的查找，但是在判断BST的合法性或在BST中搜索、增加、删除元素均可以基于这个框架，下面我们来看看。

2 判断BST的合法性

不要认为每个节点与左右节点存在如下的大小关系

• root.left.val < root.val < root.right.val

就认定该二叉搜索树是合法的，如下图。

应该添加辅助函数与函数参数列表min和max，代码如下：

boolean isValidBST(TreeNode root) {
    return isValidBST(root, null, null);
}

/* 限定以 root 为根的子树节点必须满足 max.val > root.val > min.val */
boolean isValidBST(TreeNode root, TreeNode min, TreeNode max) {
    // base case
    if (root == null) return true;
    // 若 root.val 不符合 max 和 min 的限制，说明不是合法 BST
    if (min != null && root.val <= min.val) return false;
    if (max != null && root.val >= max.val) return false;
    // 限定左子树的最大值是 root.val，右子树的最小值是 root.val
    return isValidBST(root.left, min, root) 
        && isValidBST(root.right, root, max);
}

3 在BST中搜索一个数

虽然是二叉搜索树类型的题目，但是其实二叉搜索树也是二叉树，需要首先明确函数的功能和返回值的意义。

• 在以根节点为root的树下寻找值为x的节点。
• 如果找到，返回该节点，否则返回null。

明确了上述两点后，结合二叉搜索树的框架就可以很容易写出如下代码：

TreeNode traverse(TreeNode root, int x) {
    if (root == null) return null;
    if (root.val == x) {
        return root;
    }
    if (root.val > x) {
        return traverse(root.right, x);
    }
    if (root.val < x) {
        return traverse(root.right, x);
    }
}

4 在BST中添加一个数

需要首先明确函数的功能和返回值的意义

• 在以根节点为root的树中插入值为x的节点。

• 返回根节点。

• 此时临界条件是root == null。

TreeNode traverse(TreeNode root, int x) {
    if (root == null) return new TreeNode(x);
    if (root.val > x) {
        root.right = traverse(root.right, x);
    }
    if (root.val < x) {
        root.left = traverse(root.right, x);
    }
    return root;
}

5 在BST中删除一个数

在BST中删除一个节点时，有如下三种情况：

• 删除结点左右子树均为null，则直接删除。
• 删除结点子树只有一侧为null，则由不会null的结点替换删除的节点。
• 两侧均不为null，需要左侧最大节点或者右侧最小节点替换之。

下面来明确一下函数的功能

• 函数功能：删除以root为根节点下值为key的结点，返回root。

因此，就明确了：

• 如果key在root的左边，则root.left = deleteNode(root.left, key)
• 如果key在root的右边，则root.right = deleteNode(root.right, key)
• 如果key等于root节点的值，则结合上述三种情况分别进行判断。

TreeNode deleteNode(TreeNode root, int key) {
    if (root == null) return null;
    if (root.val == key) {
        // 这两个 if 把情况 1 和 2 都正确处理了
        if (root.left == null) return root.right;
        if (root.right == null) return root.left;
        // 处理情况 3
        TreeNode minNode = getMin(root.right);
        root.val = minNode.val;
        root.right = deleteNode(root.right, minNode.val);
    } else if (root.val > key) {
        root.left = deleteNode(root.left, key);
    } else if (root.val < key) {
        root.right = deleteNode(root.right, key);
    }
    return root;
}

TreeNode getMin(TreeNode node) {
    // BST 最左边的就是最小的
    while (node.left != null) node = node.left;
    return node;
} 

6 参考

• 手把手刷二叉搜索树（第二期)