LeetCode 814

递归修剪二叉树

原创于 2018-07-22 17:49:04 发布 · 254 阅读

0 ·

CC 4.0 BY-SA版权

java 同时被 2 个专栏收录

3 篇文章

订阅专栏

leetcode

2 篇文章

订阅专栏

本文介绍了一种通过递归算法来修剪二叉树的方法，具体实现为：遍历二叉树并移除所有值为0的叶子节点，最终返回修剪后的树。

递归判断叶子节点是否为0，若叶子节点为0则修剪返回null

class Solution {
    public TreeNode pruneTree(TreeNode root) {
        if(root == null) return null;
        help(root);
        return root;
    }
    public TreeNode help(TreeNode root){
        if(root == null) return null;        
        root.left=help(root.left);
        root.right=help(root.right);
        if(root.val == 0&&root.left==null&&root.right==null) return null;
        return root;
    }
}