Leetcode 814. 二叉樹剪枝
阿新 • • 發佈:2018-08-27
left 描述 刪除 給定 esc turn -c class val
題目鏈接
https://leetcode-cn.com/problems/binary-tree-pruning/description/
題目描述
給定二叉樹根結點 root ,此外樹的每個結點的值要麽是 0,要麽是 1。
返回移除了所有不包含 1 的子樹的原二叉樹。
( 節點 X 的子樹為 X 本身,以及所有 X 的後代。)
示例1:
輸入: [1,null,0,0,1]
輸出: [1,null,0,null,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]
說明:
- 給定的二叉樹最多有 100 個節點。
- 每個節點的值只會為 0 或 1 。
題解
遞歸遍歷,直到葉子節點,然後再判斷是否需要刪除,如果需要刪除,就返回null,如果不是就直接返回。
代碼
/** * 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; } }
Leetcode 814. 二叉樹剪枝