​力扣解法汇总814-二叉树剪枝

 目录链接：

力扣编程题-解法汇总_分享+记录-CSDN博客

GitHub同步刷题项目：

https://github.com/September26/java-algorithms

原题链接：力扣

描述：

给你二叉树的根结点 root ，此外树的每个结点的值要么是 0 ，要么是 1 。

返回移除了所有不包含 1 的子树的原二叉树。

节点 node 的子树为 node 本身加上所有 node 的后代。

示例 1：

输入：root = [1,null,0,0,1]
输出：[1,null,0,null,1]
解释：
只有红色节点满足条件“所有不包含 1 的子树”。 右图为返回的答案。
示例 2：

输入：root = [1,0,1,0,0,0,1]
输出：[1,null,1,null,1]
示例 3：

输入：root = [1,1,0,1,1,0,1,0]
输出：[1,1,0,1,1,null,1]
 

提示：

树中节点的数目在范围 [1, 200] 内
Node.val 为 0 或 1

来源：力扣（LeetCode）
链接：https://leetcode.cn/problems/binary-tree-pruning
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

解题思路：

* 解题思路：
* 递归的思路，
* 每次递归的时候，分别去判断left,right,自身都不包含1，如果都不包含，则返回false。反之返回true。
* 上一层接收到false时，则删除此节点。

代码：

public class Solution814 {

    public TreeNode pruneTree(TreeNode root) {
        boolean b = have1Node(root);
        return !b ? null : root;
    }

    /**
     * root中包含1则返回true，否则为false
     *
     * @param root
     * @return
     */
    public boolean have1Node(TreeNode root) {
        TreeNode left = root.left;
        TreeNode right = root.right;
        boolean result = false;
        if (left != null) {
            if (have1Node(left)) {
                result = true;
            } else {
                root.left = null;
            }
        }
        if (right != null) {
            if (have1Node(right)) {
                result = true;
            } else {
                root.right = null;
            }
        }
        return result || root.val == 1;
    }
}