LeetCode二叉树剪枝

最新推荐文章于 2023-11-01 12:04:23 发布

liyatjj

最新推荐文章于 2023-11-01 12:04:23 发布

阅读量377

点赞数

分类专栏： Java LeetCode 文章标签： leetcode 剪枝算法

本文链接：https://blog.youkuaiyun.com/liyatjj/article/details/127018489

版权

Java 同时被 2 个专栏收录

107 篇文章

订阅专栏

LeetCode

100 篇文章

订阅专栏

该博客介绍了如何解决LeetCode上的剑指OfferII047问题，即删除所有值为0且没有子节点的子树。提供的解决方案是通过递归方法遍历二叉树，当遇到值为0且左右子树都为空的节点时将其设为null，从而实现剪枝。代码中展示了具体的递归实现过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

剑指 Offer II 047. 二叉树剪枝

给定一个二叉树根节点 root ，树的每个节点的值要么是 0，要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。

节点 node 的子树为 node 本身，以及所有 node 的后代。

输入: [1,null,0,0,1]
输出: [1,null,0,null,1]

来源：LeetCode

首先要将题目分析对，题目的意思是，当某个节点值为0，并且其左孩子为null，且右孩子也为null时才会进行剪枝。

采用递归的方法，递归结束的条件就是上面三点；同时也要将相应节点变为null。

如果某个节点是null，那就直接返回null即可。

遍历的时候，先递归遍历其左子树，再递归遍历其右子树。

class Solution {
    public TreeNode pruneTree(TreeNode root) {
        if(root==null){
            return null;
        }
        root.left=pruneTree(root.left);
        root.right=pruneTree(root.right);
        if(root.left==null&&root.right==null&&root.val==0){
            root=null;
        }
        return root;
    }
}