leetcode450. Delete Node in a BST

题目要求

Given a root node reference of a BST and a key, delete the node with the given key in the BST. Return the root node reference (possibly updated) of the BST.

Basically, the deletion can be divided into two stages:

Search for a node to remove.
If the node is found, delete the node.
Note: Time complexity should be O(height of tree).

Example:

root = [5,3,6,2,4,null,7]
key = 3

    5
   / \
  3   6
 / \   \
2   4   7

Given key to delete is 3. So we find the node with value 3 and delete it.

One valid answer is [5,4,6,2,null,null,7], shown in the following BST.

    5
   / \
  4   6
 /     \
2       7

Another valid answer is [5,2,6,null,4,null,7].

    5
   / \
  2   6
   \   \
    4   7

假设有一棵二叉搜索树,现在要求从二叉搜索树中删除指定值,使得删除后的结果依然是一棵二叉搜索树。

思路和代码

二叉搜索树的特点是,对于树中的任何一个节点,一定满足大于其所有左子节点值,小于所有其右子节点值。当删除二叉搜索树中的一个节点时,一共有三种场景:

  1. 该该节点为叶节点,此时无需进行任何操作,直接删除该节点即可
  2. 该节点只有一个子树,则将唯一的直接子节点替换掉当前的节点即可
  3. 该节点既有做左子节点又有右子节点。这时候有两种选择,要么选择左子树的最大值,要么选择右子树的最小值填充至当前的节点,再递归的在子树中删除对应的最大值或是最小值。

对每种情况的图例如下:

1. 叶节点
    5
   / \
  2   6
   \   \
    4   7 (删除4)
结果为:
    5
   / \
  2   6
       \
        7
        
2. 只有左子树或是只有右子树
    5
   / \
  3   6
 / \   \
2   4   7(删除6)
结果为
    5
   / \
  3   6
 / \   
2   4   

3. 既有左子树又有右子树
    6
   / \
  3   7
 / \   \
2   5   8 (删除6)
   /
  4
首先找到6的左子树中的最大值为5,将5填充到6的位置
    5
   / \
  3   7
 / \   \
2   5   8 (删除5)
   /
  4
接着递归的在左子树中删除5,此时5满足只有一个子树的场景,因此直接用子树替换即可
    5
   / \
  3   7
 / \   \
2   4   8 (删除5)

代码如下:

    public TreeNode deleteNode(TreeNode cur, int key) {
        if(cur == null) return null;
        else if(cur.val == key) {
            if(cur.left != null && cur.right != null) {
                TreeNode left = cur.left;
                while(left.right != null) {
                    left = left.right;
                }
                cur.val = left.val;
                cur.left = deleteNode(cur.left, left.val);
            }else if(cur.left != null) {
                return cur.left;
            }else if(cur.right != null){
                return cur.right;
            }else {
                return null;
            }
        }else if(cur.val > key) {
            cur.left = deleteNode(cur.left, key);
        }else {
            cur.right = deleteNode(cur.right, key);
        }
        return cur;
    }

心怀远方,负重前行

2.7k 声望
2k 粉丝
0 条评论
推荐阅读
从零开始JVM(一):初探JVM运行时数据区域
最近重新开始阅读《深入了解Java虚拟机》这本书,就想着用一个系列文章来记录和分享自己的心得。为什么要说”重新“呢?是因为这本书我在多年前就买了,中间也曾翻来覆去的看过。这个”翻来覆去“可以说是非常的生动...

raledong2阅读 1.9k

封面图
Java8的新特性
Java语言特性系列Java5的新特性Java6的新特性Java7的新特性Java8的新特性Java9的新特性Java10的新特性Java11的新特性Java12的新特性Java13的新特性Java14的新特性Java15的新特性Java16的新特性Java17的新特性Java...

codecraft32阅读 27.5k评论 1

一文彻底搞懂加密、数字签名和数字证书!
微信搜索🔍「编程指北」,关注这个写干货的程序员,回复「资源」,即可获取后台开发学习路线和书籍来源:个人CS学习网站:[链接]前言这本是 2020 年一个平平无奇的周末,小北在家里刷着 B 站,看着喜欢的 up 主视...

编程指北71阅读 33.7k评论 20

Java11的新特性
Java语言特性系列Java5的新特性Java6的新特性Java7的新特性Java8的新特性Java9的新特性Java10的新特性Java11的新特性Java12的新特性Java13的新特性Java14的新特性Java15的新特性Java16的新特性Java17的新特性Java...

codecraft28阅读 19.3k评论 3

Java5的新特性
Java语言特性系列Java5的新特性Java6的新特性Java7的新特性Java8的新特性Java9的新特性Java10的新特性Java11的新特性Java12的新特性Java13的新特性Java14的新特性Java15的新特性Java16的新特性Java17的新特性Java...

codecraft13阅读 21.8k

Java9的新特性
Java语言特性系列Java5的新特性Java6的新特性Java7的新特性Java8的新特性Java9的新特性Java10的新特性Java11的新特性Java12的新特性Java13的新特性Java14的新特性Java15的新特性Java16的新特性Java17的新特性Java...

codecraft20阅读 15.4k

Java13的新特性
Java语言特性系列Java5的新特性Java6的新特性Java7的新特性Java8的新特性Java9的新特性Java10的新特性Java11的新特性Java12的新特性Java13的新特性Java14的新特性Java15的新特性Java16的新特性Java17的新特性Java...

codecraft17阅读 11.2k

心怀远方,负重前行

2.7k 声望
2k 粉丝
宣传栏