leetcode1110. 删点成林-CSDN博客

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

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number[]} to_delete
 * @return {TreeNode[]}
 */
var delNodes = function(root, to_delete) {
    let result = [];
    let needDeleted = new Set(to_delete);
    if(!needDeleted.has(root.val)){
        result.push(root);
    }
    tools(root,needDeleted,{left:root});
    return result;
    function tools(node,needDeleted,p){
        if(!node) return;
        if(needDeleted.has(node.val)){
            node.left && !needDeleted.has(node.left.val) && result.push(node.left);
            node.right && !needDeleted.has(node.right.val) && result.push(node.right);
        }
        node.left && tools(node.left,needDeleted,node);
        node.right && tools(node.right,needDeleted,node);
        if(needDeleted.has(node.val)){
            node === p.left ? p.left = null : p.right = null;
        }
    }
};