LeetCode二叉树剪枝

剑指 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;
    }
}