目录链接:
力扣编程题-解法汇总_分享+记录-CSDN博客
GitHub同步刷题项目:
GitHub - September26/java-algorithms: 算法题汇总,包含牛客,leetCode,lintCode等网站题目的解法和代码,以及完整的mode类,甚至链表代码生成工具都有提供。
原题链接:力扣
描述:
给出二叉树的根节点 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
、各不相同的值。
解题思路:
/**
* 思路:
* search方法的返回值决定是否要删除当前节点。
* true代表删除,false代表不删除。
* 递归方法中,首先对左右子节点进行递归遍历,如果返回的是删除当前节点,则把对应的左右节点设置为nullptr.
* 然后查找当前节点的val是否在set中,则需要把左右节点加入到list中。但是左右节点需要判空,如果为空,则说明此节点其实已经被删掉了,则也不需要加入到list中。
*/
代码:
#include <iostream>
#include <map>
#include <list>
#include <vector>
#include <set>
#include "Solution1110.h"
#include "../model/TreeNode.h"bool search(vector<TreeNode *> &list, set<int> set, TreeNode *root)
{if (root == nullptr){return false;}if (search(list, set, root->left)){root->left = nullptr;}if (search(list, set, root->right)){root->right = nullptr;}if (set.find(root->val) != set.end()){if (root->left != nullptr){list.push_back(root->left);}if (root->right != nullptr){list.push_back(root->right);}return true;}return false;
}/*** 思路:* search方法的返回值决定是否要删除当前节点。* true代表删除,false代表不删除。***/
vector<TreeNode *> Solution1110::delNodes(TreeNode *root, vector<int> &to_delete)
{vector<TreeNode *> list;std::set<int> mySet;for (int i : to_delete){mySet.insert(i);}if (!search(list, mySet, root)){list.push_back(root);}return list;
}