【树-中等】1325. 删除给定值的叶子节点

最新推荐文章于 2025-01-07 21:38:43 发布

菜菜2022

最新推荐文章于 2025-01-07 21:38:43 发布

阅读量121

点赞数

CC 4.0 BY-SA版权

分类专栏：刷题 # leetcode

本文链接：https://blog.youkuaiyun.com/kz_java/article/details/119699009

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

该博客主要讨论如何从二叉树中递归地删除所有值等于给定目标的叶子节点，直到无法继续删除。提供了两种不同的解决方案，通过深度优先搜索（DFS）策略来实现这一操作，确保树的正确性并保持效率。

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

【题目】
给你一棵以 root 为根的二叉树和一个整数 target ，请你删除所有值为 target 的叶子节点。

注意，一旦删除值为 target 的叶子节点，它的父节点就可能变成叶子节点；如果新叶子节点的值恰好也是 target ，那么这个节点也应该被删除。

也就是说，你需要重复此过程直到不能继续删除。

示例 1：

输入：root = [1,2,3,2,null,2,4], target = 2
输出：[1,null,3,null,4]
解释：
上面左边的图中，绿色节点为叶子节点，且它们的值与 target 相同（同为 2 ），它们会被删除，得到中间的图。
有一个新的节点变成了叶子节点且它的值与 target 相同，所以将再次进行删除，从而得到最右边的图。

示例 2：

输入：root = [1,3,3,3,2], target = 3
输出：[1,3,null,null,2]

示例 3：

输入：root = [1,2,null,2,null,2], target = 2
输出：[1]
解释：每一步都删除一个绿色的叶子节点（值为 2）。

示例 4：

输入：root = [1,1,1], target = 1
输出：[]

示例 5：

输入：root = [1,2,3], target = 1
输出：[1,2,3]

提示：

1 <= target <= 1000
每一棵树最多有 3000 个节点。
每一个节点值的范围是 [1, 1000] 。

【代码】
在这里插入图片描述

class Solution:
    def dfs(self,root,target,fa,loc):
        if not root:
            return
        if not root.left and not root.right and root.val==target:
            self.flag=True
            if loc==-1:
                fa.left=None
            else:
                fa.right=None
        else:
            self.dfs(root.left,target,root,-1)
            self.dfs(root.right,target,root,1)

    def removeLeafNodes(self, root: TreeNode, target: int) -> TreeNode:
        self.flag=True
        while self.flag:
            self.flag=False
            self.dfs(root,target,None,0)
            if not root.left and not root.right and root.val==target:
                root=None
                self.flag=False
        return root

【方法2】
在这里插入图片描述

class Solution:
    def dfs(self,root,target):
        if not root:
            return None
        root.left=self.dfs(root.left,target)
        root.right=self.dfs(root.right,target)
        if root.val==target and not root.left and not root.right:
            root=None
        return root
    def removeLeafNodes(self, root: TreeNode, target: int) -> TreeNode:
        return self.dfs(root,target)