LCR 047. 二叉树剪枝 和 leetCode 1110. 删点成林 + 递归 + 图解

2023-11-29 09:15

本文主要是介绍LCR 047. 二叉树剪枝 和 leetCode 1110. 删点成林 + 递归 + 图解,希望对大家解决编程问题提供一定的参考价值,需要的开发者们随着小编来一起学习吧!

给定一个二叉树 根节点 root ,树的每个节点的值要么是 0,要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。节点 node 的子树为 node 本身,以及所有 node 的后代。


示例 1:

输入: [1,null,0,0,1]
输出: [1,null,0,null,1] 
解释: 
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。


示例 2:

输入: [1,0,1,0,0,0,1]
输出: [1,null,1,null,1]
解释: 


示例 3:

输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]
解释: 


(1)解法一: 

class Solution {
public:bool dfs(TreeNode* node) {if(node==nullptr) return false;bool left = dfs(node->left);bool right = dfs(node->right);// 叶子节点且值为0 执行删除if(left==false && right==false && node->val == 0) return false;// 非叶子节点左孩子返回false,将其删除,具体操作为node->left = nullptrif(left==false) node->left = nullptr;// 非叶子节点右孩子返回false,将其删除,具体操作为node->right = nullptrif(right==false) node->right = nullptr; return true;}TreeNode* pruneTree(TreeNode* root) {return dfs(root)?root:nullptr;}
};

 (2)解法二:

class Solution {
public:int dfs(TreeNode* node) {if(node==nullptr) return 0;int left = dfs(node->left);int right = dfs(node->right);if(left==0) node->left=nullptr;if(right==0) node->right=nullptr;return left+right+node->val;}TreeNode* pruneTree(TreeNode* root) {int ans = dfs(root);if(ans==0) return nullptr;return root;}
};

(3)解法三:

class Solution {
public:TreeNode* pruneTree(TreeNode* root) {if(root == nullptr) return nullptr;root->left = pruneTree(root->left);root->right = pruneTree(root->right);if(root->left==nullptr && root->right==nullptr && root->val == 0) { // 如果叶子节点的值为0就删除该节点return nullptr;}return root;}
};

leetCode 1110. 删点成林 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]]

解法一:“递归”的思路可以分为「递」「归」两个部分。

  • 「递」的思路:向下传参,然后不断进行处理(类似前序遍历
  • 「归」的思路:先遍历到底,然后向上传参(类似后序遍历

本题中采用「归」的思路,需要解决两个问题:

  • 1.向上传递的参数是什么
  • 2.在当前节点如何进行处理

思路和分析:

  • 1.如果该节点(node)不需要被删除,那么返回 node 即可;如果需要被删除,那么返回 null 即可
  • 2.在处理需要被删除的节点(node)时,删除前先判断其有无左右子树,若有就将其左(右)子树加入ans中,再返回 null 即可

class Solution {
public:vector<TreeNode *> ans;unordered_set<int> delSet;bool dfs(TreeNode *node) {if(node==nullptr) return false;bool left = dfs(node->left);bool right = dfs(node->right);if(delSet.count(node->val)) {if(left) ans.push_back(node->left);if(right) ans.push_back(node->right);return false;}if(left==false) node->left = nullptr;if(right==false) node->right = nullptr; return node;}vector<TreeNode *> delNodes(TreeNode *root, vector<int> &to_delete) {for(const auto &a:to_delete) {delSet.insert(a);}if (dfs(root)) ans.push_back(root);return ans;}
};

解法二:

class Solution {
public:vector<TreeNode *> ans;unordered_set<int> delSet;TreeNode * dfs(TreeNode *node) {if(node==nullptr) return nullptr;node->left = dfs(node->left);node->right = dfs(node->right);if(delSet.count(node->val)) {if(node->left) ans.push_back(node->left);if(node->right) ans.push_back(node->right);return nullptr; // 相当于删除节点}return node;// 没有删除}vector<TreeNode *> delNodes(TreeNode *root, vector<int> &to_delete) {for(const auto &a:to_delete) {delSet.insert(a);}if (dfs(root)) ans.push_back(root);return ans;}
};

解法三(来自灵茶山艾府的题解),和解法二思路差不多:文字总结来自灵神的文章:1110. 删点成林 - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289131/he-shi-ji-lu-da-an-pythonjavacgo-by-endl-lpcd/

class Solution {vector<TreeNode *> ans;unordered_set<int> s;TreeNode *dfs(TreeNode *node) {if (node == nullptr) return nullptr;node->left = dfs(node->left);node->right = dfs(node->right);if (!s.count(node->val)) return node;if (node->left) ans.push_back(node->left);if (node->right) ans.push_back(node->right);return nullptr;}public:vector<TreeNode *> delNodes(TreeNode *root, vector<int> &to_delete) {for (int x: to_delete) s.insert(x);if (dfs(root)) ans.push_back(root);return ans;}
};

总结:考虑清楚两个点

  •     1.对叶子节点被删,如何处理?直接丢掉节点
  •     2.对非叶子节点被删,如何处理?把子节点加入结果,当前节点丢掉即可

最后判断树根是否为空,非空则加入ans

推荐和参考文章:

1110. 删点成林 - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289131/he-shi-ji-lu-da-an-pythonjavacgo-by-endl-lpcd/1110. 删点成林 - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289170/er-cha-shu-de-hou-xu-bian-li-cong-xia-wa-0y4l/1110. 删点成林 - 力扣(LeetCode)icon-default.png?t=N7T8https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289224/shan-dian-cheng-lin-cai-yong-di-gui-zhon-9jsj/

这篇关于LCR 047. 二叉树剪枝 和 leetCode 1110. 删点成林 + 递归 + 图解的文章就介绍到这儿,希望我们推荐的文章对编程师们有所帮助!



http://www.chinasem.cn/article/432263

相关文章

哈希leetcode-1

目录 1前言 2.例题  2.1两数之和 2.2判断是否互为字符重排 2.3存在重复元素1 2.4存在重复元素2 2.5字母异位词分组 1前言 哈希表主要是适合于快速查找某个元素(O(1)) 当我们要频繁的查找某个元素,第一哈希表O(1),第二,二分O(log n) 一般可以分为语言自带的容器哈希和用数组模拟的简易哈希。 最简单的比如数组模拟字符存储,只要开26个c

usaco 1.3 Prime Cryptarithm(简单哈希表暴搜剪枝)

思路: 1. 用一个 hash[ ] 数组存放输入的数字,令 hash[ tmp ]=1 。 2. 一个自定义函数 check( ) ,检查各位是否为输入的数字。 3. 暴搜。第一行数从 100到999,第二行数从 10到99。 4. 剪枝。 代码: /*ID: who jayLANG: C++TASK: crypt1*/#include<stdio.h>bool h

leetcode-24Swap Nodes in Pairs

带头结点。 /*** Definition for singly-linked list.* public class ListNode {* int val;* ListNode next;* ListNode(int x) { val = x; }* }*/public class Solution {public ListNode swapPairs(L

leetcode-23Merge k Sorted Lists

带头结点。 /*** Definition for singly-linked list.* public class ListNode {* int val;* ListNode next;* ListNode(int x) { val = x; }* }*/public class Solution {public ListNode mergeKLists

C++ | Leetcode C++题解之第393题UTF-8编码验证

题目: 题解: class Solution {public:static const int MASK1 = 1 << 7;static const int MASK2 = (1 << 7) + (1 << 6);bool isValid(int num) {return (num & MASK2) == MASK1;}int getBytes(int num) {if ((num &

hdu1010 奇偶剪枝

恰好t时间到达 import java.io.BufferedReader;import java.io.InputStream;import java.io.InputStreamReader;import java.io.PrintWriter;import java.math.BigInteger;import java.util.Arrays;import

【每日一题】LeetCode 2181.合并零之间的节点(链表、模拟)

【每日一题】LeetCode 2181.合并零之间的节点(链表、模拟) 题目描述 给定一个链表,链表中的每个节点代表一个整数。链表中的整数由 0 分隔开,表示不同的区间。链表的开始和结束节点的值都为 0。任务是将每两个相邻的 0 之间的所有节点合并成一个节点,新节点的值为原区间内所有节点值的和。合并后,需要移除所有的 0,并返回修改后的链表头节点。 思路分析 初始化:创建一个虚拟头节点

C语言 | Leetcode C语言题解之第393题UTF-8编码验证

题目: 题解: static const int MASK1 = 1 << 7;static const int MASK2 = (1 << 7) + (1 << 6);bool isValid(int num) {return (num & MASK2) == MASK1;}int getBytes(int num) {if ((num & MASK1) == 0) {return

leetcode105 从前序与中序遍历序列构造二叉树

根据一棵树的前序遍历与中序遍历构造二叉树。 注意: 你可以假设树中没有重复的元素。 例如,给出 前序遍历 preorder = [3,9,20,15,7]中序遍历 inorder = [9,3,15,20,7] 返回如下的二叉树: 3/ \9 20/ \15 7   class Solution {public TreeNode buildTree(int[] pr

【JavaScript】LeetCode:16-20

文章目录 16 无重复字符的最长字串17 找到字符串中所有字母异位词18 和为K的子数组19 滑动窗口最大值20 最小覆盖字串 16 无重复字符的最长字串 滑动窗口 + 哈希表这里用哈希集合Set()实现。左指针i,右指针j,从头遍历数组,若j指针指向的元素不在set中,则加入该元素,否则更新结果res,删除集合中i指针指向的元素,进入下一轮循环。 /*** @param