Leetcode 二叉树剪枝

java solution

java 复制代码
class Solution {
    public TreeNode pruneTree(TreeNode root) {
        //递归后序遍历来解决此题
        if(root == null) return null;

        //先剪枝左右子树
        root.left = pruneTree(root.left);
        root.right = pruneTree(root.right);

        //然后处理当前节点

        if(root.val == 0 && root.left == null && root.right == null) {
            return null;
        }
        //否则,保留当前节点
        return root;
    }
}

这部分代码片段,为什么判断条件不是 root.val == 0 && root.left.val == 0 && root.right.val == 0

复制代码
if (root.val == 0 && root.left == null && root.right == null) {
            return null;
}

root.left == nullroot.right == null 的判断是 判断子树是否已经被剪掉(即不存在),而不是判断子节点的值是否为 0。


👇 详细解释如下:

root.left == 0 && root.right == 0 ------ 错误的写法
  • root.leftroot.rightTreeNode 类型,不是整数,不能直接用 == 0 比较。
  • 即使你访问的是值(比如 root.left.val == 0),也不够,因为:
    • 即使 val == 0,只判断这一层还不够,整个子树里也可能包含 1,那这个子树就不能剪掉。
root.left == null && root.right == null ------ 正确的做法
  • 表示 当前节点的左子树和右子树都已经被递归剪掉了(要么原本是空的,要么在递归过程中被剪为 null)。
  • 然后再判断当前节点的值:
    • 如果是 0,那么这个节点及其所有后代都不包含 1,就应该被剪掉,返回 null
    • 如果是 1,即使左右子树都为空,也要保留这个节点。

🔄 递归的核心思想:

在后序遍历中,我们是先把左子树和右子树递归"剪干净"后,才处理当前节点。

所以:

java 复制代码
root.left = pruneTree(root.left);
root.right = pruneTree(root.right);

之后,root.leftroot.right 已经是处理完的结果 ------ 如果某一子树里没有 1,它就已经是 null

这时候判断:

java 复制代码
if (root.val == 0 && root.left == null && root.right == null)

意味着:

当前节点是 0,并且它的子树中没有 1 ------ 满足"整棵子树没有 1"的条件,因此需要剪掉。

相关推荐
圆头猫爹3 分钟前
第34次CCF-CSP认证第4题,货物调度
c++·算法·动态规划
秋说6 分钟前
【PTA数据结构 | C语言版】出栈序列的合法性
c语言·数据结构·算法
用户403159863966320 分钟前
多窗口事件分发系统
java·算法
用户403159863966323 分钟前
ARP 缓存与报文转发模拟
java·算法
hi0_642 分钟前
03 数组 VS 链表
java·数据结构·c++·笔记·算法·链表
aPurpleBerry42 分钟前
hot100 hot75 栈、队列题目思路
javascript·算法
卷福同学2 小时前
【AI编程】AI+高德MCP不到10分钟搞定上海三日游
人工智能·算法·程序员
mit6.8242 小时前
[Leetcode] 预处理 | 多叉树bfs | 格雷编码 | static_cast | 矩阵对角线
算法
皮卡蛋炒饭.2 小时前
数据结构—排序
数据结构·算法·排序算法
??tobenewyorker3 小时前
力扣打卡第23天 二叉搜索树中的众数
数据结构·算法·leetcode