LeetCode-Python-1110. 删点成林（树 + DFS）

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

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

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

示例 1：

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

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

提示：

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

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/delete-nodes-and-return-forest
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：

题目要求把所有新的树的root全部添加到数组里然后返回。

一个node成为新的树的root必须满足两个条件：

1. 它爹被删了

2. 它自己不能被删

然后基础的 DFS 就可以解决这道题。

时间复杂度：O（N），因为每个节点最多遍历一次

空间复杂度：O（N），to_delete 转成集合以及最坏情况下递归开栈都是O（N）

# 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 delNodes(self, root: Optional[TreeNode], to_delete: List[int]) -> List[TreeNode]:
        to_delete = set(to_delete)
        res = [] 
        def dfs(node, node_is_new_root):
            if not node:
                return
            left = node.left
            right = node.right
            if node_is_new_root and node.val not in to_delete:
                res.append(node)
            if left and left.val in to_delete:
                node.left = None
            if right and right.val in to_delete:
                node.right = None
             
            dfs(left, node.val in to_delete)
            dfs(right, node.val in to_delete)
        dfs(root, True)
        return res