LeetCode算法日记：814. 二叉树剪枝

happykoi

于 2022-08-03 21:04:48 发布

阅读量209

点赞数

CC 4.0 BY-SA版权

文章标签：算法 leetcode 剪枝

本文链接：https://blog.csdn.net/happykoi/article/details/126148978

该博客介绍了如何使用深度优先搜索（DFS）解决814号LeetCode问题——二叉树剪枝。通过递归地修剪不包含1的子树，保留含有1的节点，实现了对二叉树的修剪。示例展示了不同输入情况下二叉树修剪后的结果。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

814. 二叉树剪枝

日期：2022/8/3

题目描述：给你二叉树的根结点 root ，此外树的每个结点的值要么是 0 ，要么是 1 。

返回移除了所有不包含 1 的子树的原二叉树。

节点 node 的子树为 node 本身加上所有 node 的后代。

示例：

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

输入：root = [1,0,1,0,0,0,1]
输出：[1,null,1,null,1]

输入：root = [1,1,0,1,1,0,1,0]
输出：[1,1,0,1,1,null,1]

思路：

dfs

代码+解析：

class Solution {
public:
    TreeNode* pruneTree(TreeNode* root) {
        if(root == nullptr) return 0;
        
        root->left = pruneTree(root->left);
        root->right = pruneTree(root->right);

        if(!(root->left || root->right || root->val)){
            return nullptr;
        }
        return root;
    }
};