leetcode python3 简单题112. Path Sum

做一只AI小能手

于 2020-04-27 23:05:45 发布

阅读量154

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/qq_37285386/article/details/105801409

版权

leetcode 专栏收录该内容

59 篇文章 0 订阅

订阅专栏

1.编辑器

我使用的是win10+vscode+leetcode+python3
环境配置参见我的博客：
链接

2.第一百一十二题

（1）题目
英文：
Given a binary tree and a sum, determine if the tree has a root-to-leaf path such that adding up all the values along the path equals the given sum.

Note: A leaf is a node with no children.

中文：
给定一个二叉树和一个目标和，判断该树中是否存在根节点到叶子节点的路径，这条路径上所有节点值相加等于目标和。

说明: 叶子节点是指没有子节点的节点。

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/path-sum

（2）解法
① 循环法
（耗时：52ms，内存：15.7M）

class Solution:
    def hasPathSum(self, root: TreeNode, sum: int) -> bool:
        if not root:
            return False
        stack = [(root,sum)]
        while stack:
            node,val = stack.pop()
            if not node.left and not node.right and node.val == val:
                return True
            if node.right:
                stack.append((node.right,val-node.val))
            if node.left:
                stack.append((node.left,val-node.val))
        return False

② DFS
（耗时：44ms，内存：15.5M）

class Solution:
    def hasPathSum(self, root: TreeNode, sum: int) -> bool:
        if not root: return False
        if not root.left and not root.right and sum - root.val == 0:return True
        return self.hasPathSum(root.left, sum - root.val) or self.hasPathSum(root.right, sum - root.val)