前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >二叉树:删除节点

二叉树:删除节点

作者头像
灰子学技术
发布2020-08-21 16:13:01
7300
发布2020-08-21 16:13:01
举报
文章被收录于专栏:灰子学技术灰子学技术

算法:

1.后驱算法:

代码语言:javascript
复制
/*
递归解法:
1.找到需要删除的节点
2.删除的节点只有右子树或者左子树,直接将右子树或者左子树的根节点当作这个删除的节点
3.删除的节点左右子树都存在的情况下,左子树的最大节点也叫做前驱当作删除节点,
或者将右子树的最小节点也就称作后驱当作删除节点。
*/

2.前驱算法:

代码语言:javascript
复制
/*
递归解法:
1.找到需要删除的节点
2.删除的节点只有右子树或者左子树,直接将右子树或者左子树的根节点当作这个删除的节点
3.删除的节点左右子树都存在的情况下,左子树的最大节点也叫做前驱当作删除节点,
或者将右子树的最小节点也就称作后驱当作删除节点。
*/

题目:

https://leetcode-cn.com/problems/delete-node-in-a-bst/

后驱算法:

代码语言:javascript
复制
func deleteNode(root *TreeNode, key int) *TreeNode {
    if root == nil {
        return nil
    }
    if key < root.Val {
        root.Left = deleteNode(root.Left,key)
        return root
    }
    if key > root.Val {
        root.Right = deleteNode(root.Right,key)
        return root
    }
    // 这里通过递归已经找到了要删除的节点,此时因为递归的原因还是root这个指针
    if root.Right == nil {
        return root.Left
    }
    if root.Left == nil {
        return root.Right
    }
    // 后驱的方法来替换当前节点
    min := root.Right
    for min.Left != nil {
        min = min.Left
    }
    root.Val = min.Val // 替换需要删除的节点的数值,这里就是复制
    root.Right = deleteMin(root.Right) // 这里把右子树中移动过来的那个最小节点删除掉
    return root
}
func deleteMin(root *TreeNode) *TreeNode{
    // 左子树不在的话,表示这个节点就是要删除的最小节点
    // 存在两种情况,一:这个节点就是叶子节点,直接通过赋值为nil, 来当作删除节点。
    // 二:这个节点没有左子树,只有右子树,这样的话,需要将右子树替换成该节点
    if root.Left == nil { 
        right := root.Right
        root.Right = nil 
        return right
    }
    root.Left = deleteMin(root.Left) // 左子树一直在的话,就一直通过左子树去找最小节点
    return root
}

前驱算法

代码语言:javascript
复制
func deleteNode(root *TreeNode, key int) *TreeNode {
    if root == nil {
        return nil
    }
    if key < root.Val {
        root.Left = deleteNode(root.Left,key)
        return root
    }
    if key > root.Val {
        root.Right = deleteNode(root.Right,key)
        return root
    }
    // 这里通过递归已经找到了要删除的节点,此时因为递归的原因还是root这个指针
    if root.Right == nil {
        return root.Left
    }
    if root.Left == nil {
        return root.Right
    }
    // 前驱的方法来替换当前节点
    max := root.Left
    for max.Right != nil {
        max = max.Right
    }
    root.Val = max.Val
    root.Left = deleteMax(root.Left)
    return root
}

func deleteMax(root *TreeNode) *TreeNode {
    if root.Right == nil {
        left := root.Left
        root.Left = nil 
        return left
    }
    root.Right = deleteMax(root.Right)
    return root
}
/*
递归解法:
1.找到需要删除的节点
2.删除的节点只有右子树或者左子树,直接将右子树或者左子树的根节点当作这个删除的节点
3.删除的节点左右子树都存在的情况下,左子树的最大节点也叫做前驱当作删除节点,
或者将右子树的最小节点也就称作后驱当作删除节点。
*/
本文参与 腾讯云自媒体分享计划,分享自微信公众号。
原始发表:2020-08-17,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 灰子学技术 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体分享计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档