【Leetcode_总结】814. 二叉树剪枝 - python

最新推荐文章于 2022-08-27 11:18:24 发布

Maka_uir

最新推荐文章于 2022-08-27 11:18:24 发布

阅读量618

点赞数

CC 4.0 BY-SA版权

分类专栏： Leetcode 文章标签：树

本文链接：https://blog.youkuaiyun.com/maka_uir/article/details/86981667

这篇博客介绍了LeetCode中814题的解决方案，即如何移除所有不含1的子树。文章通过递归方法，详细阐述了判断和执行剪枝操作的思路，并提供了相应的Python代码实现。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Q：

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

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

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

示例1:
输入: [1,null,0,0,1]
输出: [1,null,0,null,1]
 
解释: 
只有红色节点满足条件“所有不包含 1 的子树”。
右图为返回的答案。

链接：https://leetcode-cn.com/problems/binary-tree-pruning/description/

思路：使用递归，判断是否满足减枝的条件

代码：

# Definition for a binary tree node.
# class TreeNode(object):
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None

class Solution(object):
    def pruneTree(self, root):
        """
        :type root: TreeNode
        :rtype: TreeNode
        """
        if self.shouldRemove(root):
            return None
        return root
    def shouldRemove(self,nd):