剑指 Offer II 047. 二叉树剪枝
给定一个二叉树 根节点 root ,树的每个节点的值要么是 0,要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。
节点 node 的子树为 node 本身,以及所有 node 的后代。
输入: [1,null,0,0,1]
输出: [1,null,0,null,1]
首先要将题目分析对,题目的意思是,当某个节点值为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;
}
}