Leetcode 814. Binary Tree Pruning

最新推荐文章于 2024-11-05 15:14:03 发布

weixin_30246221

最新推荐文章于 2024-11-05 15:14:03 发布

阅读量73

点赞数

文章标签：数据结构与算法

原文链接：http://www.cnblogs.com/zywscq/p/10754162.html

版权

本文介绍了一种使用深度优先搜索(DFS)进行树结构剪枝的方法，通过递归遍历树节点并移除所有值为0的子树，从而实现对二叉树的有效修剪。代码中采用了Python语言，并利用了functools模块的lru_cache装饰器来提高效率。

摘要由CSDN通过智能技术生成

dfs

要点是这一句:

return node.val==1 or node.left or node.right

完整代码:

# Definition for a binary tree node.
# class TreeNode:
#     def __init__(self, x):
#         self.val = x
#         self.left = None
#         self.right = None
import functools
class Solution:
    @functools.lru_cache()
    def pruneTree(self, root: TreeNode) -> TreeNode:
        def dfs(node:TreeNode)->bool:
            if not node:
                return False
            if not dfs(node.left):
                node.left=None
            if not dfs(node.right):
                node.right=None
            return node.val==1 or node.left or node.right
        if not root:
            return
        return root if dfs(root) else None