题目描述:
给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
说明: 叶子节点是指没有子节点的节点。
解题思路:
注意:需要分情况讨论,当根节点为空时,不存在路径。
自顶向下遍历,每次遍历一条边的时候,sum减去这条变的值。
# Definition for a binary tree node.
#给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。
class TreeNode(object):
def __init__(self, x):
self.val = x
self.left = None
self.right = None
class Solution:
def hasPathSum(self, root: TreeNode, sum: int) -> bool:
if not root: return False
t = sum - root.val
if not root.left and not root.right:
return t == 0
return self.hasPathSum(root.left, t) or self.hasPathSum(root.right, t)
n1 = TreeNode(4)
n2 = TreeNode(2)
n3 = TreeNode(7)
n4 = TreeNode(1)
n5 = TreeNode(3)
n1.left = n2
n1.right = n3
n2.left = n4
n2.right = n5
s = Solution()
result = s.hasPathSum(n1,10)
print(result)