(每日一道算法题)二叉树剪枝

814. 二叉树剪枝 - 力扣(LeetCode)

算法思路:深度优先的后序遍历

​核心逻辑​​:自底向上检查每棵子树,只有全零叶子节点才被移除。

​后序遍历顺序​​:先递归处理左右子树,再判断当前节点

  • 确保子树已被修剪,当前节点基于最新子树状态决策

​删除条件​​:仅当节点同时满足

  • 节点值为 0
  • 无左子树(left == null
  • 无右子树(right == null)
    时才被移除(返回 null 断开父节点引用)

​为何这样设计?​

  • 非叶子节点即使值为 0,只要连接有效子树(含 1)就不能删除(如示例1的 0 节点)
  • 后序遍历保证从叶子向上处理,避免重复判断

代码实现

java 复制代码
class Solution {
    public TreeNode pruneTree(TreeNode root) {
        if (root == null) return null;
        
        // 先递归处理子树
        root.left = pruneTree(root.left);
        root.right = pruneTree(root.right);
        
        // 当且仅当是叶子节点且值为0时移除
        if (root.val == 0 && root.left == null && root.right == null) {
            return null;
        }
        return root;
    }
}

示例推演

java 复制代码
原始树:        递归后:        返回结果:
    1            1              1
      \            \              \
       0    →       0     →         0
      / \             \               \
     0   1             1               1
  1. 递归到左叶节点 0
    满足 val=0 且无子树 → 返回 null(移除)
    → 父节点 0 的左子树置为 null
  2. ​处理父节点 0
    0 但右子树含 1 → 保留
  3. ​根节点 1
    1 直接保留,最终形成新结构

关键步骤解析

​深度递归​
root.left = pruneTree(root.left) 先深入到左子树底部
root.right = pruneTree(root.right) 再深入到右子树底部
(形成自底向上的处理链)

​精准剪枝​
if (root.val == 0 && root.left == null && root.right == null)

仅当当前节点是值为 0 的叶子节点时才移除,保留两种情况:

  • 值为 1 的节点(无论是否叶子)
  • 值为 0 但连接有效子树的中间节点

​时间复杂度​

  • O(n):每个节点仅访问一次(n 为节点数)
  • 空间复杂度 O(h):递归栈深度(h 为树高,最坏 O(n))

总结

通过​​后序遍历+叶子节点精准判断​​实现高效剪枝:

  • 适用场景:树型数据处理中移除无效分支(如零值配置、空目录清理)
  • 扩展思考:若修改为"保留至少含一个 1 的子树",代码逻辑不变(本解法已实现)
  • 工程启示:递归深度优先搜索是处理树型结构的利器,以 O(n) 时间完成复杂裁剪
相关推荐
Blossom.1187 分钟前
基于深度学习的图像分类:使用DenseNet实现高效分类
人工智能·深度学习·目标检测·机器学习·分类·数据挖掘·迁移学习
屁股割了还要学32 分钟前
【C语言进阶】柔性数组
c语言·开发语言·数据结构·c++·学习·算法·柔性数组
草莓熊Lotso32 分钟前
【LeetCode刷题指南】--有效的括号
c语言·数据结构·其他·算法·leetcode·刷题
Alla T41 分钟前
【通识】算法案例
算法
Electrolux44 分钟前
你敢信,不会点算法没准你赛尔号都玩不明白
前端·后端·算法
天天开心(∩_∩)1 小时前
代码随想录算法训练营第三十一天
算法
qq_513970441 小时前
力扣 hot100 Day55
算法·leetcode
赴3352 小时前
机器学习 KNN 算法,鸢尾花案例
人工智能·机器学习·sklearn·knn·鸢尾花
不绝1913 小时前
ARPG开发流程第一章——方法合集
算法·游戏·unity·游戏引擎
盼小辉丶3 小时前
图机器学习(20)——欺诈交易检测算法
人工智能·机器学习·图机器学习