112. 路径总和

112. 路径总和

题目

使用广度优先搜索的方式,记录从根节点到当前节点的路径和。
使用两个队列,分别存储将要遍历的节点,以及根节点到这些节点的路径和即可。

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

class Solution:
    def hasPathSum(self, root: TreeNode, sum: int) -> bool:
        if root==None:
            return False
        nodestack = [root]
        sumstack = [root.val]
        while nodestack:
            p = nodestack.pop(0)
            cursum = sumstack.pop(0)
            
            if p.left:
                cur_left_sum = cursum + p.left.val
                nodestack.append(p.left)
                sumstack.append(cur_left_sum)
            if p.right:
                cur_right_sum = cursum + p.right.val
                nodestack.append(p.right)
                sumstack.append(cur_right_sum)

            if not p.left and not p.right:
                if cursum==sum:
                    return True
        return False
©著作权归作者所有,转载或内容合作请联系作者
【社区内容提示】社区部分内容疑似由AI辅助生成,浏览时请结合常识与多方信息审慎甄别。
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

友情链接更多精彩内容