剑指 Offer II 047. 二叉树剪枝

题目描述

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

题目解析

思路：

• 通过递归遍历左子树和右子树，剪枝的条件是该节点值为0且没有左子树和右子树，找出符合条件的节点return 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) {
            return null;
        }
        return root;
    }
}