LeetCode 814. 二叉树剪枝

dan_zhoudan

于 2020-07-21 09:14:41 发布

阅读量164

点赞数

分类专栏： LeetCode 文章标签：二叉树剪枝算法 leetcode 二叉树剪枝

本文链接：https://blog.youkuaiyun.com/zhouth94/article/details/107480510

版权

LeetCode 专栏收录该内容

14 篇文章

订阅专栏

本文详细解析了LeetCode上的一道经典题目——二叉树剪枝。通过对题目的深入分析，阐述了如何通过递归算法高效地移除所有不包含1的子树，使读者能够快速理解并掌握这一算法技巧。

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

1、题目链接

https://leetcode-cn.com/problems/binary-tree-pruning/

给定二叉树根结点 root ，此外树的每个结点的值要么是 0，要么是 1。

返回移除了所有不包含 1 的子树的原二叉树。

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

2、分析

二叉树的大部分题目都可以使用递归来思考。

当前访问节点的左、右子树都为null（也就是为叶子节点），且当前节点的值为0，则需要删除。也就相当于删除值为0的叶子结点。

3、代码

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode(int x) { val = x; }
 * }
 */
class Solution {
    //若当前节点的左、右子树都为空或为0（叶子节点），且该节点值为0，则删
    public TreeNode pruneTree(TreeNode root) {
        if(root == null) return root;
        return helper(root);
    }

    TreeNode helper(TreeNode root) {
        if(root == null) return null;
        TreeNode l = helper(root.left);
        TreeNode r = helper(root.right);

        if(l == null && r == null && root.val == 0) return null;

        root.left = l;
        root.right = r;

        return root;
    }

}