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"的条件,因此需要剪掉。

相关推荐
仙俊红30 分钟前
LeetCode每日一题,20250914
算法·leetcode·职场和发展
风中的微尘7 小时前
39.网络流入门
开发语言·网络·c++·算法
西红柿维生素8 小时前
JVM相关总结
java·jvm·算法
ChillJavaGuy10 小时前
常见限流算法详解与对比
java·算法·限流算法
sali-tec10 小时前
C# 基于halcon的视觉工作流-章34-环状测量
开发语言·图像处理·算法·计算机视觉·c#
你怎么知道我是队长11 小时前
C语言---循环结构
c语言·开发语言·算法
艾醒11 小时前
大模型面试题剖析:RAG中的文本分割策略
人工智能·算法
纪元A梦13 小时前
贪心算法应用:K-Means++初始化详解
算法·贪心算法·kmeans
_不会dp不改名_13 小时前
leetcode_21 合并两个有序链表
算法·leetcode·链表
mark-puls14 小时前
C语言打印爱心
c语言·开发语言·算法