【亡羊补牢】挑战数据结构与算法 第38期 LeetCode 450. 删除二叉搜索树中的节点(二叉树)【详解】
发布日期:2021-06-29 14:34:17 浏览次数:3 分类:技术文章

本文共 3657 字,大约阅读时间需要 12 分钟。

仰望星空的人,不应该被嘲笑

题目描述

给定一个二叉搜索树的根节点 root 和一个值 key,删除二叉搜索树中的 key 对应的节点,并保证二叉搜索树的性质不变。返回二叉搜索树(有可能被更新)的根节点的引用。

一般来说,删除节点可分为两个步骤:

首先找到需要删除的节点;

如果找到了,删除它。
说明: 要求算法时间复杂度为 O(h),h 为树的高度。

示例:

root = [5,3,6,2,4,null,7]key = 3    5   / \  3   6 / \   \2   4   7给定需要删除的节点值是 3,所以我们首先找到 3 这个节点,然后删除它。一个正确的答案是 [5,4,6,2,null,null,7], 如下图所示。    5   / \  4   6 /     \2       7另一个正确答案是 [5,2,6,null,4,null,7]。    5   / \  2   6   \   \    4   7

来源:力扣(LeetCode)

链接:https://leetcode-cn.com/problems/delete-node-in-a-bst
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

解题思路

对于这道题,我们必须先了解一下二叉搜索树(BST)的性质,如下:

BST性质

  • 中序遍历是升序
  • left小于当前节点,right大于当前节点
  • 左子树、右子树也要是BST

了解了性质之后,我们知道要查找对应 key 值,可以与根节点进行比较,如果小于根节点,直接去左子树找就好了,如果大于根节点,直接去右子树找就好了。

而对于刚好等于根节点的话,我拿着大佬的图解来看看几种情况:

第一种情况,如果删除节点仅有右孩子,直接指向右孩子

第二种情况,如果删除节点仅有左孩子,直接指向左孩子
第三种情况,如果删除节点左右孩子都有,那么我们按照题意可以有两种删除方法:

① 找到要删除节点左子树的最右边的节点,即前驱的最大值(由BST性质得到)替换当前 root 节点,然后删除这个前驱

② 找到要删除节点右子树的最左边的节点,即后继最小值(由BST性质得到)替换当前 root 节点,然后删除这个后继

参考 大佬图解

代码提供了两种解法,对于前两种情况处理是一样的,只是对于最后一种情况,可以删除前驱的最大值,或者删除后继的最小值。

删除前驱的最大值代码

/** * Definition for a binary tree node. * function TreeNode(val, left, right) { *     this.val = (val===undefined ? 0 : val) *     this.left = (left===undefined ? null : left) *     this.right = (right===undefined ? null : right) * } *//** * @param {TreeNode} root * @param {number} key * @return {TreeNode} */var deleteNode = function (root, key) {
if (!root) return null; // 判断值是否小于root,小于走左子树,大于走右子树 if (key < root.val) {
root.left = deleteNode(root.left, key); } else if (key > root.val) {
root.right = deleteNode(root.right, key); } else {
// 1.如果删除节点没有左右子树,直接删除即可 if (!root.left && !root.right) {
root = null; // 2.如果删除节点仅有左孩子,直接指向左孩子 } else if (root.left && !root.right) {
root = root.left; // 3.如果删除节点仅有右孩子,直接指向右孩子 } else if (!root.left && root.right) {
root = root.right; } else {
// 4.如果左右孩子都有,本代码采用方式是将前驱的最大值替换root的值 let last = root.left; while (last.right) {
last = last.right; } root.val = last.val; // 然后删除这个前驱最大值节点即可 root.left = deleteNode(root.left, last.val); } } return root;};

删除后继的最小值代码

/** * Definition for a binary tree node. * function TreeNode(val, left, right) { *     this.val = (val===undefined ? 0 : val) *     this.left = (left===undefined ? null : left) *     this.right = (right===undefined ? null : right) * } *//** * @param {TreeNode} root * @param {number} key * @return {TreeNode} */var deleteNode = function (root, key) {
if (!root) return null; // 判断值是否小于root,小于走左子树,大于走右子树 if (key < root.val) {
root.left = deleteNode(root.left, key); } else if (key > root.val) {
root.right = deleteNode(root.right, key); } else {
// 1.如果删除节点没有左右子树,直接删除即可 if (!root.left && !root.right) {
root = null; // 2.如果删除节点仅有左孩子,直接指向左孩子 } else if (root.left && !root.right) {
root = root.left; // 3.如果删除节点仅有右孩子,直接指向右孩子 } else if (!root.left && root.right) {
root = root.right; } else {
// 4.如果左右孩子都有,本代码采用方式是将后继的最小值替换root的值 let last = root.right; while (last.left) {
last = last.left; } root.val = last.val; // 然后删除这个后继最小值节点即可 root.right = deleteNode(root.right, last.val); } } return root;};

最后

文章产出不易,还望各位小伙伴们支持一波!

往期精选:

小伙伴们可以在Issues中提交自己的解题代码,🤝 欢迎Contributing,可打卡刷题,Give a ⭐️ if this project helped you!

,方便小伙伴阅读玩耍~

学如逆水行舟,不进则退

转载地址:https://chocolate.blog.csdn.net/article/details/108781484 如侵犯您的版权,请留言回复原文章的地址,我们会给您删除此文章,给您带来不便请您谅解!

上一篇:【亡羊补牢】挑战数据结构与算法 第39期 LeetCode 501. 二叉搜索树中的众数(二叉树)
下一篇:【亡羊补牢】挑战数据结构与算法 第37期 LeetCode 108. 将有序数组转换为二叉搜索树(二叉树)

发表评论

最新留言

第一次来,支持一个
[***.219.124.196]2024年04月25日 03时20分22秒