一刷237-剑指 Offer II 047. 二叉树剪枝(同:814. 二叉树剪枝)

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

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

在这里插入图片描述

示例:
输入: [1,null,0,0,1]
输出: [1,null,0,null,1] 
解释: 
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。

在这里插入图片描述

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

在这里插入图片描述

输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]
---------------
提示:
二叉树的节点个数的范围是 [1,200]
二叉树节点的值只会是 01
-----------------
思路:
虽然这道题看似是让我们去剪枝,但其实仔细考虑下条件,我们从叶子节点网上逆推:

左子树为空
右子树为空
node.val == 0
当满足以上三个条件时,将该叶子节点删除(node = null),即可。
通过深度优先搜索每个节点的子节点,递归返回,就是最终的答案了。
这里需要注意下root节点为空时,特殊判断。
-----------------
/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    public TreeNode pruneTree(TreeNode root) {
        if (root == null) return root;//特判
        root.left = pruneTree(root.left);//左子节点递归
        root.right = pruneTree(root.right);//右子节点递归
        if (root.val == 0 && root.left == null && root.right == null) {
            root = null;//剪枝操作
        }
        return root;
    }
}

LC

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值