814. 二叉树剪枝

给定二叉树根结点 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;
         }
    }
}




发布了89 篇原创文章 · 获赞 3 · 访问量 5738

猜你喜欢

转载自blog.csdn.net/Hqxcsdn/article/details/104678217