[LeetCode] 450. Delete Node in a BST @ python

最新推荐文章于 2022-12-19 13:45:28 发布

原创最新推荐文章于 2022-12-19 13:45:28 发布 · 192 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#leetcode

leetcode 专栏收录该内容

35 篇文章

订阅专栏

本文介绍了一种在二叉搜索树(BST)中删除特定节点的方法，确保删除后树的结构仍然符合BST特性。核心思路在于，当目标节点存在右子树时，用其右子树中的最小值替代并进行递归删除。

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

一．题目：
删除二叉树中指定一节点，并调整二叉树，使结果仍是BST。
二．解题思路：
我们应该逐个访问树上的节点，如果某个节点的值正好是目标值，那么：
若它没有右子树，那么直接用它的左孩子替换它；
若它有右子树．那么将它的值和它右子树中的最小节点的值调换。
代码如下：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def deleteNode(self, root, key):
        """
        :type root: TreeNode
        :type key: int
        :rtype: TreeNode
        """
        if not root:
            return None
        if root.val == key:
            if not root.right:
                return root.left
            else:
                right = root.right
                while right.left:
                    right = right.left
                root.val,right.val = right.val,root.val
        root.left = self.deleteNode(root.left,key)
        root.right = self.deleteNode(root.right,key)
        return root