《剑指 Offer》专项突破版 - 面试题 47 : 二叉树剪枝(C++ 实现)

题目链接LCR 047. 二叉树剪枝 - 力扣(LeetCode)

题目

一棵二叉树的所有节点的值要么是 0 要么是 1,请剪除该二叉树中所有节点的值全都是 0 的子树。例如,在剪除下图 (a) 中二叉树中所有节点值都为 0 的子树之后的结果如下图 (b) 所示。

分析

首先分析哪些子树会被剪除,哪些子树不能被剪除。在上图 (a) 的二叉树中,以第 2 层第 1 个节点为根节点的子树的 3 个节点的值都是 0,因此整棵二叉树都被剪除。以第 2 层第 2 个节点为根节点的子树中有一个节点的值是 1(第 3 层第 4 个节点),因此这棵子树不能删除,但是它的左子树(根节点为第 3 层第 3 个节点)只有一个节点并且值为 0,因此它的左子树可以被剪除。

下面总结什么样的节点可以被删除。首先,这个节点的值应该是 0。其次,如果它有子树,那么它的子树的所有节点的值都为 0。也就是说,如果一个节点可以被删除,那么它的子树的所有节点都可以被删除。例如,在上图 (a) 的二叉树中,第 2 层第 1 个节点可以被删除,它的子树中的所有节点(第 3 层第 1 个节点和第 2 个节点)也可以被删除。

由此发现,后序遍历 最适合用来解决这个问题。如果用后序遍历的顺序遍历到某个节点,那么它的左右子树的节点一定已经遍历过了。每遍历到一个节点,就要确定它是否有左右子树,如果左右子树都是空的,并且节点的值是 0,那么就可以删除这个节点。

代码实现

cpp 复制代码
class Solution {
public:
    TreeNode* pruneTree(TreeNode* root) {
        if (!root)
            return nullptr;
        
        root->left = pruneTree(root->left);
        root->right = pruneTree(root->right);
        if (!root->left && !root->right && root->val == 0)
            return nullptr;
        
        return root;
    }
};
相关推荐
YuforiaCode5 分钟前
第十二届蓝桥杯 2021 C/C++组 直线
c语言·c++·蓝桥杯
我真的不会C9 分钟前
QT窗口相关控件及其属性
开发语言·qt
CodeCraft Studio9 分钟前
Excel处理控件Aspose.Cells教程:使用 Python 在 Excel 中进行数据验
开发语言·python·excel
火柴盒zhang15 分钟前
websheet之 编辑器
开发语言·前端·javascript·编辑器·spreadsheet·websheet
景天科技苑22 分钟前
【Rust】Rust中的枚举与模式匹配,原理解析与应用实战
开发语言·后端·rust·match·enum·枚举与模式匹配·rust枚举与模式匹配
知来者逆23 分钟前
计算机视觉——速度与精度的完美结合的实时目标检测算法RF-DETR详解
图像处理·人工智能·深度学习·算法·目标检测·计算机视觉·rf-detr
阿让啊28 分钟前
C语言中操作字节的某一位
c语言·开发语言·数据结构·单片机·算法
এ᭄画画的北北29 分钟前
力扣-160.相交链表
算法·leetcode·链表
椰羊~王小美33 分钟前
LeetCode -- Flora -- edit 2025-04-25
java·开发语言
草莓啵啵~1 小时前
搜索二叉树-key的搜索模型
数据结构·c++