LintCode:删除二叉查找树的节点

最新推荐文章于 2018-07-15 16:06:48 发布

原创最新推荐文章于 2018-07-15 16:06:48 发布 · 660 阅读

0 ·

CC 4.0 BY-SA版权

lintcode 同时被 2 个专栏收录

139 篇文章

订阅专栏

python

116 篇文章

订阅专栏

本文介绍了一种从二叉查找树中删除指定值节点的方法。通过递归搜索找到目标节点，并根据节点的不同情况（没有子节点、仅有一个子节点、有两个子节点）采取不同的删除策略。对于有两个子节点的情况，采用右子树中的最小值节点替代当前节点。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

LintCode:删除二叉查找树的节点

"""
Definition of TreeNode:
class TreeNode:
    def __init__(self, val):
        self.val = val
        self.left, self.right = None, None
"""
class Solution:
    """
    @param root: The root of the binary search tree.
    @param value: Remove the node with given value.
    @return: The root of the binary search tree after removal.
    """    
    def removeNode(self, root, value):
        if not root:
            return
        if root.val > value:
            root.left = self.removeNode(root.left, value)
        elif root.val < value:
            root.right = self.removeNode(root.right, value)
        elif root.left and root.right:
            root.val = self.findMin(root.right)
            self.removeNode(root.right, root.val)
        else:
            return root.left if root.left else root.right
        return root

    def findMin(self, root):
        while root.left:
            root = root.left
        return root.val