-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy path112.路径总和.py
More file actions
63 lines (60 loc) · 1.61 KB
/
112.路径总和.py
File metadata and controls
63 lines (60 loc) · 1.61 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
#
# @lc app=leetcode.cn id=112 lang=python
#
# [112] 路径总和
#
# https://leetcode-cn.com/problems/path-sum/description/
#
# algorithms
# Easy (48.08%)
# Likes: 186
# Dislikes: 0
# Total Accepted: 34.4K
# Total Submissions: 71.4K
# Testcase Example: '[5,4,8,11,null,13,4,7,2,null,null,null,1]\n22'
#
# 给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
#
# 说明: 叶子节点是指没有子节点的节点。
#
# 示例:
# 给定如下二叉树,以及目标和 sum = 22,
#
# 5
# / \
# 4 8
# / / \
# 11 13 4
# / \ \
# 7 2 1
#
#
# 返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。
#
#
# @lc code=start
# 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 hasPathSum(self, root, sum):
"""
:type root: TreeNode
:type sum: int
:rtype: bool
"""
if not root:return False
stack=[(root,sum)]
while stack:
node,cursum=stack.pop()
if node.left==None and node.right==None and cursum==node.val:
return True
if node.left:
stack.append((node.left,cursum-node.val))
if node.right:
stack.append((node.right,cursum-node.val))
return False
# @lc code=end