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)解法一:

cpp 复制代码
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 = nullptr
        if(left==false) node->left = nullptr;
        // 非叶子节点右孩子返回false,将其删除,具体操作为node->right = nullptr
        if(right==false) node->right = nullptr; 
        return true;
    }
    TreeNode* pruneTree(TreeNode* root) {
        return dfs(root)?root:nullptr;
    }
};

(2)解法二:

cpp 复制代码
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)解法三:

cpp 复制代码
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即可

cpp 复制代码
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;
    }
};

解法二:

cpp 复制代码
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)https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289131/he-shi-ji-lu-da-an-pythonjavacgo-by-endl-lpcd/

cpp 复制代码
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)https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289131/he-shi-ji-lu-da-an-pythonjavacgo-by-endl-lpcd/1110. 删点成林 - 力扣(LeetCode)https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289170/er-cha-shu-de-hou-xu-bian-li-cong-xia-wa-0y4l/1110. 删点成林 - 力扣(LeetCode)https://leetcode.cn/problems/delete-nodes-and-return-forest/solutions/2289224/shan-dian-cheng-lin-cai-yong-di-gui-zhon-9jsj/

相关推荐
Lenyiin32 分钟前
02.06、回文链表
数据结构·leetcode·链表
爪哇学长35 分钟前
双指针算法详解:原理、应用场景及代码示例
java·数据结构·算法
爱摸鱼的孔乙己37 分钟前
【数据结构】链表(leetcode)
c语言·数据结构·c++·链表·csdn
Dola_Pan39 分钟前
C语言:数组转换指针的时机
c语言·开发语言·算法
繁依Fanyi1 小时前
简易安卓句分器实现
java·服务器·开发语言·算法·eclipse
烦躁的大鼻嘎1 小时前
模拟算法实例讲解:从理论到实践的编程之旅
数据结构·c++·算法·leetcode
C++忠实粉丝1 小时前
计算机网络socket编程(4)_TCP socket API 详解
网络·数据结构·c++·网络协议·tcp/ip·计算机网络·算法
用户37791362947552 小时前
【循环神经网络】只会Python,也能让AI写出周杰伦风格的歌词
人工智能·算法
福大大架构师每日一题2 小时前
文心一言 VS 讯飞星火 VS chatgpt (396)-- 算法导论25.2 1题
算法·文心一言
EterNity_TiMe_2 小时前
【论文复现】(CLIP)文本也能和图像配对
python·学习·算法·性能优化·数据分析·clip