【1110. 删点成林】

来源：力扣（LeetCode）

描述：

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

方法：深度优先搜索

思路

题目给定一棵树 root，树的每个节点都有一个各不相同的值。并且给定一个数组 to_delete，包含需要删除的节点值。返回删除所有的 to_delete 中的节点后，剩余的树的集合。

可以利用深度优先搜索来遍历每一个节点，定义函数 dfs，输入是参数是某个节点 node 和这个节点是否为潜在的新的根节点 is_root。函数中，首先判断这个节点是否要被删除，如果是，那么它的两个子节点（如果有的话）便成为了潜在的根节点。如果这个节点的值不在 to_delete 中并且 is_root 为 true，那么这个节点便成为了一个新的根节点，需要把它放入结果数组中。同时也要对它的两个子节点进行同样的操作。dfs 的返回值为更新后的 node。

对根节点调用一次 dfs，返回新的根节点数组即可。

代码：

class Solution {public:vector<TreeNode*> delNodes(TreeNode* root, vector<int>& to_delete) {unordered_set<int> to_delete_set(to_delete.begin(), to_delete.end());vector<TreeNode *> roots;function<TreeNode *(TreeNode *, bool)> dfs = [&](TreeNode* node, bool is_root) -> TreeNode * {if (node == nullptr) {return nullptr;}bool deleted = to_delete_set.count(node->val) ? true : false;node->left = dfs(node->left, deleted);node->right = dfs(node->right, deleted);if (deleted) {return nullptr;} else {if (is_root) {roots.emplace_back(node);}return node;}};dfs(root, true);return roots;}
};

执行用时：16ms, 在所有 C++ 提交中击败了92.74%的用户
内存消耗：24.6 MB, 在所有 C++ 提交中击败了77.82%的用户
复杂度分析
时间复杂度：O(n)，其中 n 是树的节点数。
空间复杂度：O(n)，其中 n 是树的节点数。
author：LeetCode-Solution

【1110. 删点成林】相关推荐

Leetcode 1110.删点成林（Delete Nodes And Return Forest）
Leetcode 1110.删点成林 1 题目描述(Leetcode题目链接) 给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节 ...
PigyChan_LeetCode 1110. 删点成林
1110. 删点成林难度中等给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的 ...
LeetCode 刷题系列 -- 1110. 删点成林
给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的集合). 返回森林中的每棵树.你 ...
1110. 删点成林
给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的集合). 返回森林中的每棵树.你 ...
leetCode 1110 删点成林（树，后序遍历）
题目链接:点击查看题目描述: 给定一个整数二叉树和一些整数,求删掉这些整数对应的节点后,剩余的子树. 输入输出: 输入:root = [1,2,3,4,5,6,7], to_delete = [3, ...
力扣解法汇总1110. 删点成林
目录链接: 力扣编程题-解法汇总_分享+记录-CSDN博客 GitHub同步刷题项目: GitHub - September26/java-algorithms: 算法题汇总,包含牛客,leetCod ...
LeetCode 1110. 删点成林（二叉树递归）
1. 题目给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的集合). 返回森林中 ...
leetcode 1110. 删点成林
题目描述: 给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的集合). 返回森林中 ...
1110 删点成林
题目描述: 给出二叉树的根节点 root,树上每个节点都有一个不同的值. 如果节点值在 to_delete 中出现,我们就把该节点从树上删去,最后得到一个森林(一些不相交的树构成的集合). 返回森林中 ...
LeetCode 1110.删点成林
主页有其他数据结构内容(持续更新中) 难度:Medium 代码: /*** Definition for a binary tree node.* struct TreeNode {* int val ...

【1110. 删点成林】

【1110. 删点成林】相关推荐

最新文章

热门文章