【树-中等】1110. 删点成林

最新推荐文章于 2023-05-30 07:15:09 发布

原创最新推荐文章于 2023-05-30 07:15:09 发布 · 101 阅读

0 ·

CC 4.0 BY-SA版权

leetcode 同时被 2 个专栏收录

714 篇文章

订阅专栏

刷题

643 篇文章

订阅专栏

该博客介绍了一个二叉树问题，给定一个二叉树的根节点和一个待删除节点值的列表，任务是删除这些节点并返回形成的森林。示例展示了输入输出情况，并给出了解决方案，通过深度优先搜索（DFS）遍历树并删除指定节点。

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

【题目】
给出二叉树的根节点 root，树上每个节点都有一个不同的值。

如果节点值在 to_delete 中出现，我们就把该节点从树上删去，最后得到一个森林（一些不相交的树构成的集合）。

返回森林中的每棵树。你可以按任意顺序组织答案。

示例：

输入：root = [1,2,3,4,5,6,7], to_delete = [3,5]
输出：[[1,2,null,4],[6],[7]]

提示：

树中的节点数最大为 1000。
每个节点都有一个介于 1 到 1000 之间的值，且各不相同。
to_delete.length <= 1000
to_delete 包含一些从 1 到 1000、各不相同的值。

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

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def dfs(self,root,loc,fa,to_delete):
        if not root:
            return
        self.dfs(root.left,-1,root,to_delete)
        self.dfs(root.right,1,root,to_delete)
        if root.val in to_delete:
            if fa:
                if loc==1:
                    fa.right=None
                elif loc==-1:
                    fa.left=None                
            if root.left:
                self.ans.append(root.left)
            if root.right:
                self.ans.append(root.right)
            
    def delNodes(self, root: TreeNode, to_delete: List[int]) -> List[TreeNode]:
        if root and root.val in to_delete:
            self.ans=[]
        else:
            self.ans=[root]
        self.dfs(root,0,None,to_delete)
        return self.ans