Leetcode 1110.删点成林（Delete Nodes And Return Forest）

Leetcode 1110.删点成林

1 题目描述（Leetcode题目链接）

  给出二叉树的根节点 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、各不相同的值。

2 题解

  深度优先搜索，使用一个辅助队列来存，详见注释。

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

class Solution:
    def delNodes(self, root: TreeNode, to_delete: List[int]) -> List[TreeNode]:
        to_delete = set(to_delete)
        self.res = []
        self.queue = collections.deque([root])
        def dfs(root):
            if not root:
                return False				
            if root.val in to_delete:       # 如果root是要删除的节点则将其左右孩子入队
                self.queue.append(root.left)
                self.queue.append(root.right)
                return True					# True 表示root是要删除的节点
            if dfs(root.left):				# 如果root的左孩子是要删除的节点，则左子树置空
                root.left = None
            if dfs(root.right):				# 如果root的右孩子是要删除的节点，则右子树置空
                root.right = None
            return False					# False 表示root不是要删除的节点

        while self.queue:
            node = self.queue.popleft()
            if not node: continue
            if not dfs(node):				# 如果root不是要删除的节点，那么就加在结果里面
                self.res.append(node)
        return self.res