leetcode删除二叉搜索树中的节点(js实现)

删除二叉搜索树中的节点

给定一个二叉搜索树的根节点 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,2,6,null,4,null,7]。

在这里插入图片描述

解题思路

递归+二叉搜索树的搜索操作
删除操作需要考虑一下三种情况:
1.如果目标节点没有子节点,则可以直接删除目标节点;
2.如果目标节点只有一个子节点,可以用其子节点作为替换;
3.如果目标节点有两个子节点,则需要用其中序序列的后继节点获前驱节点来替换,再删除该目标节点;

需要清除后继节点和前驱节点的概念:
后继节点:代表中序遍历序列的下一个节点,即比当前节点大的最小节点。先取当前节点的右节点,然后一直取该右节点的左节点,直到左节点为空,则最后指向的节点为后继节点。
前驱节点:代表中序遍历序列的前一个节点,即比当前节点小的最大节点。先取当前节点的左节点,然后一直取该左节点的右节点,直到右节点为空,则最后指向的节点为后继节点。

再结合递归搜索操作,可得:
1.如果key>root.val,说明要删除的节点在右子树,root.right = deleteNode(root.right,key)
2.如果key<root.val,说明要删除的节点在左子树,root.left = deleteNode(root.left,key)
3.如果key == root.val,则表示该节点就是我们要删除的节点,则:

  • 如果root为叶子节点,则直接删除即可,root=null

  • 如果root不是叶子节点且有左子树,则用它的前驱节点的值代替,root.val = predecessor(root)

  • 如果root不是叶子节点且有右子树,则用他的后继节点的值代替,root.val = successor(root)

4.返回root

var deleteNode = function(root, key) {
    //前驱结点
    function predecessor(root){
        root = root.left
        while(root.right!=null){
            root = root.right
        }
        return root.val
    }
    //后继结点
    function successor(root){
        root = root.right
        while(root.left!=null){
            root = root.left
        }
        return root.val
    }

    //删除操作
    if(!root){
        return root
    }
    if(key < root.val){
        root.left = deleteNode(root.left,key)
    }else if(key > root.val){
        root.right = deleteNode(root.right,key)
    }else{
        if(root.left == null && root.right == null){//key为叶子结点
            root = null
        }else if(root.left != null){//key不为叶子结点且有左子树
            root.val = predecessor(root)
            root.left = deleteNode(root.left,root.val)
        }else if(root.right != null){//key不为叶子结点且有右子树
            root.val = successor(root)
            root.right = deleteNode(root.right,root.val)
        }
    }
    return root
};

在这里插入图片描述

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值