力扣---1110. 删点成林

给出二叉树的根节点 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
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

利用一个变量来存储某个节点的父节点，再用一个变量 isHead 来存储下一个不需要删除的节点是否应该是头结点。

显然，当某个节点为被删除节点后，他的子树节点即有可能是头结点（需要注意连续删除的情况）。

class Solution {
    public List<TreeNode> delNodes(TreeNode root, int[] to_delete) {
        List<TreeNode> list = new ArrayList<>();
        Set<Integer> set = new HashSet<>() {{
            for (int x : to_delete) {
                add(x);
            }
        }};
        dfs(root, null, true, set, list, true);
        return list;
    }
    private void dfs (TreeNode root, TreeNode parent, boolean isLeft, Set<Integer> set, List<TreeNode> list, boolean isHead) {
        if (root == null) {
            return;
        }
//        当前节点是需要删除的节点
        if (set.contains(root.val)) {
//            父节点非空
            if (parent != null) {
                if (isLeft) {
                    parent.left = null;
                } else {
                    parent.right = null;
                }
            }
            dfs(root.left, null, true, set, list, true);
            dfs(root.right, null, false, set, list, true);
        } else { 
//            如果当前节点是头结点，则放入到 list 中。
            if (isHead) {
                list.add(root);
            }
            dfs(root.left, root,true, set, list, false);
            dfs(root.right, root, false, set, list, false);
        }
    }
}