Java实现 LeetCode 814 二叉树剪枝（遍历树）

浅浅的花香味﹌ 2023-03-13 01:55 35阅读 0赞

## 814. 二叉树剪枝 ##

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

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

( 节点 X 的子树为 X 本身，以及所有 X 的后代。)

示例1:  
输入: \[1,null,0,0,1\]  
输出: \[1,null,0,null,1\]

解释:  
只有红色节点满足条件“所有不包含 1 的子树”。  
右图为返回的答案。  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70]

示例2:  
输入: \[1,0,1,0,0,0,1\]  
输出: \[1,null,1,null,1\]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70 1]

示例3:  
输入: \[1,1,0,1,1,0,1,0\]  
输出: \[1,1,0,1,1,null,1\]  
![在这里插入图片描述][watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70 2]

说明:

给定的二叉树最多有 100 个节点。  
每个节点的值只会为 0 或 1 。

/** * 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;
            }
            root.left = pruneTree(root.left);
            root.right = pruneTree(root.right);
            if(root.left == null && root.right == null && root.val == 0) { 
                return null;
            }
            return root;
        }
    }

[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70]: /images/20230312/1b9b56f997094799af123de96d5a1142.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70 1]: /images/20230312/ea56eb93021d40d6a806a8108c943eb5.png
[watermark_type_ZmFuZ3poZW5naGVpdGk_shadow_10_text_aHR0cHM6Ly9ibG9nLmNzZG4ubmV0L2ExNDM5Nzc1NTIw_size_16_color_FFFFFF_t_70 2]: /images/20230312/a0cf05128a17465e886b19444de1304a.png