给定二叉树根结点 root ,此外树的每个结点的值要么是 0,要么是 1。
返回移除了所有不包含 1 的子树的原二叉树。
( 节点 X 的子树为 X 本身,以及所有 X 的后代。)
示例1:
输入: [1,null,0,0,1]
输出: [1,null,0,null,1]
解释:
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。
示例2:
输入: [1,0,1,0,0,0,1]
输出: [1,null,1,null,1]
示例3:
输入: [1,1,0,1,1,0,1,0]
输出: [1,1,0,1,1,null,1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-pruning
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
如果一条路径上都不含有1 则将路径删除,一开始想着深度遍历,找出路径,再对路径判断其是否含1 但是发现递归更适合本题解法
对任何一棵树 ,如果要删去他,那么有三个条件 1.自己为0 2.左子树全为0 3.右子树也全为0
所以采用递归方法。
/**
* Definition for a binary tree node.
* public class TreeNode {
* int val;
* TreeNode left;
* TreeNode right;
* TreeNode(int x) { val = x; }
* }
*/
class Solution {
public TreeNode pruneTree(TreeNode root) {
if(root==null){
return null;
}
if(haveOne(root)){
return root;
}
else
{
return null;
}
}
public boolean haveOne(TreeNode root){
if(root==null){
return false;
}
if(root.val==1){
//节点含有1 考虑其子树情况
boolean right = haveOne(root.right);
//如果右子树不含1 剪枝
if(!right){
root.right=null;
}
boolean left = haveOne(root.left);
if(!left){
root.left=null;
}
return true;
}
else{
//当前节点不含有1,如果子树都不含有1 那么返回null
boolean right = haveOne(root.right);
if(!right){
root.right=null;
}
boolean left = haveOne(root.left);
if(!left){
root.left=null;
}
// 只要有一子树含有1 那么该树就会有1
return left||right;
}
}
}