【算法】二叉搜索树的基本操作

  二叉搜索树是一棵数据有序的树，对于树上的每个节点，左子树的所有节点的数值都小于根节点，右子树的所有节点的数值都大于根节点。下面讲解几种二叉搜索树的基本操作。

1.查找操作

查找操作很简单，通过要查找的值和根节点对比，可以得到如下的查找过程：

• 根节点为空，查找失败；
• root.val == val，查找成功；
• root.val > val，到左子树查找；
• root.val < val，到右子树查找。

查找的最坏时间复杂度是 O(h)，h 是树高。查找的代码如下：

public TreeNode searchBST(TreeNode root, int val) {
    if (root == null) return null;
    if (root.val == val) {
        return root;
    } else if (root.val > val) {
        return searchBST(root.left, val);
    } else {
        return searchBST(root.right, val);
    }
}

练习：力扣700. 二叉搜索树中的搜索：https://leetcode-cn.com/problems/search-in-a-binary-search-tree/

2.插入操作

和查找思路类似，查找失败的地方就是需要插入的地方，最坏时间复杂度是 O(h)，h 是树高。但是要注意，在父节点就要提前判断是否查找失败，因为要将父节点的指针指向新插入的节点。

public TreeNode insertIntoBST(TreeNode root, int val) {
    if (root == null) {		// 如果原来的树为空，直接新建一棵树返回
        root = new TreeNode(val);
        return root;
    }
    insert(root, val);		// 原来的树不为空，进行插入操作
    return root;
}
public void insert(TreeNode root, int val) {
    if (root.left == null && root.val > val) {	// 在左孩子中查找失败，
        TreeNode node = new TreeNode(val);
        root.left = node;
        return;
    }
    if (root.right == null && root.val < val) {	// 在右孩子中查找失败
        TreeNode node = new TreeNode(val);
        root.right = node;
        return;
    }
    if (root.val > val) {	// 到左子树中查找
        insert(root.left, val);
    } else {				// 到右子树中查找
        insert(root.right, val);
    }
}

练习：力扣701. 二叉搜索树中的插入操作：https://leetcode-cn.com/problems/insert-into-a-binary-search-tree/

3.二叉树的建立

有了插入，建立就很简单，只需要不断插入即可。

public TreeNode sortedArrayToBST(int[] nums) {
    TreeNode root = new TreeNode(nums[0]);
    for (int i = 1; i < num.length; i++) {
        insert(root, nums[i]);
    }
    return root;
}

4.二叉树的删除

二叉树删除一种方式有两种方法，一是将其前驱节点（左子树中的最大节点）覆盖该节点，二是将其后继节点（右子树中的最小节点）覆盖该节点。

寻找前驱节点就是寻找左子树中的最大节点，代码如下：

public int predecessor(TreeNode root) {
    root = root.left;
    while (root.right != null) root = root.right;
    return root.val;
}

寻找后继节点就是寻找左子树中的最大节点，代码如下：

public int successor(TreeNode root) {
    root = root.right;
    while (root.left != null) root = root.left;
    return root.val;
}

有了寻找前驱和后继节点的代码，可以得到算法的思路如下：

• 如果 key > root.val，说明要删除的节点在右子树，root.right = deleteNode(root.right, key)。
• 如果 key < root.val，说明要删除的节点在左子树，root.left = deleteNode(root.left, key)。
• 如果 key == root.val，则该节点就是我们要删除的节点，则：
  • 如果该节点是叶子节点，则直接删除它：root = null。
  • 如果该节点不是叶子节点且有右节点，则用它的后继节点的值替代 root.val = successor.val，然后删除后继节点。
  • 如果该节点不是叶子节点且只有左节点，则用它的前驱节点的值替代 root.val = predecessor.val，然后删除前驱节点。
    返回 root。

public TreeNode deleteNode(TreeNode root, int key) {
    if (root == null) return null;

    if (key > root.val) {			// 要删除的节点在右子树
        root.right = deleteNode(root.right, key);
    } else if (key < root.val) {	// 要删除的节点在左子树
        root.left = deleteNode(root.left, key);
    } else {						// 该节点就是要删除的节点
        if (root.left == null && root.right == null) {	// 叶子节点直接删除
            root = null;
        } else if (root.right != null) {	// 有右孩子就用后继替换该节点，再递归删除后继
            root.val = successor(root);
            root.right = deleteNode(root.right, root.val);
        } else {							// 有左孩子就用前驱替换该节点，再递归删除前驱
            root.val = predecessor(root);
            root.left = deleteNode(root.left, root.val);
        }
    }
    return root;
}

练习：力扣450. 删除二叉搜索树中的节点：https://leetcode-cn.com/problems/delete-node-in-a-bst/
欢迎关注公众号。每天更新大数据开发面经。