2022-8-13 剑指offer-二叉树递归

剑指 Offer II 047. 二叉树剪枝

难度中等

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

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

 1 /**
 2  * Definition for a binary tree node.
 3  * public class TreeNode {
 4  *     int val;
 5  *     TreeNode left;
 6  *     TreeNode right;
 7  *     TreeNode() {}
 8  *     TreeNode(int val) { this.val = val; }
 9  *     TreeNode(int val, TreeNode left, TreeNode right) {
10  *         this.val = val;
11  *         this.left = left;
12  *         this.right = right;
13  *     }
14  * }
15  */
16 class Solution {
17     public TreeNode pruneTree(TreeNode root) {
18         if (root==null) return null;
19         if (root.val==0&&isZero(root)){
20             return null;
21         }else{
22             root.left=pruneTree(root.left);
23             root.right=pruneTree(root.right);
24             return root;
25         }
26     }
27 
28     public boolean isZero(TreeNode root){
29         if (root==null) return true;
30         if (root.val!=0) return false;
31         return isZero(root.left)&&isZero(root.right);
32     }
33 }

思路：递归调用剪枝，关键在于判断一个子树是否为全为0的树。

posted on 2022-08-13 12:16  阿ming  阅读(15)  评论(0)  编辑  收藏  举报

导航