LeetCode 每日一题——814. 二叉树剪枝

1.题目描述

814. 二叉树剪枝

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

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

节点 node 的子树为 node 本身加上所有 node 的后代。

示例 1：

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

示例 2：

输入：root = [1,0,1,0,0,0,1]
输出：[1,null,1,null,1]

示例 3：

输入：root = [1,1,0,1,1,0,1,0]
输出：[1,1,0,1,1,null,1]

2.解题思路与代码

2.1 解题思路

题目要求删除节点全是 0 的子树来完成二叉树剪枝，我们使用一个递归函数来完成。首先递归函数输入一个节点，返回以这个节点为根节点的子树是否全部都是 0 。在函数内部递归调用函数分别判断这个节点左子树和右子树是否全部为零，如果左子树全部为 0 ，则将当前节点的 left 指向 null；同样如果右子树全部为 0 ，right 指向 null，最后判断以当前节点为根节点的子树是否全部为 0 ，即判断当前节点是否为 0、左右子树是否全部为 0，并返回结果。需要注意叶节点的情况，如果叶节点值为 0 返回 true，默认删除为 0 的叶节点。

2.2 代码

class Solution {
    public TreeNode pruneTree(TreeNode root) {
        boolean process = process(root);
        if (process){
            return null;
        }
        return root;
    }

    public boolean process(TreeNode node) {
        if (node == null) {
            return true;
        }
        if (node.left == null && node.right == null) {
            return node.val == 0;
        }
        boolean left = process(node.left);
        boolean right = process(node.right);
        if (left) {
            node.left = null;
        }
        if (right) {
            node.right = null;
        }
        return node.val == 0 && left && right;
    }
}

2.3 测试结果

通过测试

3.总结

• 使用递归分别判断当前节点左右子树是否全为 0 ，如果全部为 0 ，则直接设置成 null
• 叶节点值为 0 返回 true，默认删除为 0 的叶节点