3分钟秒杀。。。
class Solution {
public TreeNode pruneTree(TreeNode root) {
dfs(root);
if(!judge(root)) {
return null;
}
return root;
}
boolean dfs(TreeNode node) {
if(node == null) {
return false;
}
if(!dfs(node.left)) {
node.left = null;
}
if(!dfs(node.right)) {
node.right = null;
}
if(!judge(node)) {
return false;
}
return true;
}
boolean judge(TreeNode node) {
if(node.val == 0 && node.left == null && node.right == null) {
return false;
}
return true;
}
}