剑指 Offer II 047. 二叉树剪枝

导读:本篇文章讲解 剑指 Offer II 047. 二叉树剪枝,希望对大家有帮助,欢迎收藏,转发!站点地址:www.bmabk.com

题目

给定一个二叉树 根节点 root ,树的每个节点的值要么是 0,要么是 1。请剪除该二叉树中所有节点的值为 0 的子树。

节点 node 的子树为 node 本身,以及所有 node 的后代。

示例 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]
解释:

提示:

二叉树的节点个数的范围是 [1,200]
二叉树节点的值只会是 0 或 1

来源:力扣(LeetCode)
链接:https://leetcode.cn/problems/pOCWxh
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。

思路和代码

后序遍历,递归

/**
 * Definition for a binary tree node.
 * public class TreeNode {
 *     int val;
 *     TreeNode left;
 *     TreeNode right;
 *     TreeNode() {}
 *     TreeNode(int val) { this.val = val; }
 *     TreeNode(int val, TreeNode left, TreeNode right) {
 *         this.val = val;
 *         this.left = left;
 *         this.right = right;
 *     }
 * }
 */
class Solution {
    // 使用后序遍历
    public TreeNode pruneTree(TreeNode root) {
        // base case
        if(root == null){
            return root;
        }
        root.left = pruneTree(root.left);
        root.right = pruneTree(root.right);
        if(root.left == null && root.right == null && root.val == 0){
            // 返回到当前调用的递归函数(把那个变成null),所以上面分别是 root.left和root.right
            return null;
        }
        // 递归是从下向前进行减的,所有这样是错的,下面先减掉的话,再来他已经是null了
        // if(root.left == 0 && root.right == 0 && root.val == 0){
        //     return null;
        // }
        return root;

    }
}

版权声明:本文内容由互联网用户自发贡献,该文观点仅代表作者本人。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如发现本站有涉嫌侵权/违法违规的内容, 请发送邮件至 举报,一经查实,本站将立刻删除。

文章由极客之音整理,本文链接:https://www.bmabk.com/index.php/post/96161.html

(0)
小半的头像小半

相关推荐

极客之音——专业性很强的中文编程技术网站,欢迎收藏到浏览器,订阅我们!