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

相关推荐
_Coin_-10 分钟前
算法训练营DAY27 第八章 贪心算法 part01
算法·贪心算法
董董灿是个攻城狮5 小时前
5分钟搞懂什么是窗口注意力?
算法
Dann Hiroaki5 小时前
笔记分享: 哈尔滨工业大学CS31002编译原理——02. 语法分析
笔记·算法
qqxhb6 小时前
零基础数据结构与算法——第四章:基础算法-排序(上)
java·数据结构·算法·冒泡·插入·选择
FirstFrost --sy8 小时前
数据结构之二叉树
c语言·数据结构·c++·算法·链表·深度优先·广度优先
森焱森8 小时前
垂起固定翼无人机介绍
c语言·单片机·算法·架构·无人机
搂鱼1145148 小时前
(倍增)洛谷 P1613 跑路/P4155 国旗计划
算法
Yingye Zhu(HPXXZYY)8 小时前
Codeforces 2021 C Those Who Are With Us
数据结构·c++·算法
无聊的小坏坏10 小时前
三种方法详解最长回文子串问题
c++·算法·回文串
长路 ㅤ   10 小时前
Java后端技术博客汇总文档
分布式·算法·技术分享·编程学习·java后端