【Leetcode Daily】814二叉树剪枝

题目浅析

• 想查看原题可以点击题目链接。

• 简单地说，就是给一个数值只有 0 和 1 的二叉树，找到其中的只有 0 的子树删掉。

思路分享

• 官方题解镇楼

  https://leetcode.cn/problems/binary-tree-pruning/solutions/1683846/er-cha-shu-jian-zhi-by-leetcode-solution-k336/

• 居然能和官方题解一摸一样，看来功力应该还是有提升的吧。

• 题目要求删除一整个子树，二叉树只能从上往下遍历，所以自然利用递归，让每个节点返回自己以及下面是否只有 0。

• 具体怎么做呢？对于边缘叶子的 0，如果返回的时候让上一层把对应的位置置空，这样删除整个为 0 的子树就转化成了一个个删除 0 的叶子节点。

代码解答（强烈建议自行解答后再看）

• 参考题解

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, val=0, left=None, right=None):
#         self.val = val
#         self.left = left
#         self.right = right
class Solution:
    def pruneTree(self, root: Optional[TreeNode]) -> Optional[TreeNode]:
        if not root:
            return None
        root.left = self.pruneTree(root.left)
        root.right = self.pruneTree(root.right)
        if root.left is None and root.right is None and root.val == 0:
            return None
        return root