题目:
给定一棵二叉树和一个总和,确定该树中是否存在根到叶的路径,这条路径的所有值相加等于给定的总和。
例如:
给定下面的二叉树和 总和 = 22
,
5 / \ 4 8 / / \ 11 13 4 / \ \ 7 2 1
返回 true
, 因为存在总和为 22 的根到叶的路径 5->4->11->2
。
分析:
- 使用dfs遍历每条路径;
- 遍历时记录根节点到当前节点之前的和;
- 当节点为叶子节点时,此时的和为该路径根节点到叶子节点和,判断该值是否为目标值。
代码:
class Solution(object):
def hasPathSum(self, root, sum):
"""
:type root: TreeNode
:type sum: int
:rtype: bool
"""
self.flag = False
def dfs(node,sumNow = 0,target = sum):
if not self.flag:
if node:
sumNow += node.val
if node.left or node.right:
dfs(node.left, sumNow, target)
dfs(node.right, sumNow, target)
else:
if sumNow == target:
self.flag = True
if root:
dfs(root)
return self.flag
思考:
- 开始将对比路径节点和与目标值对比放在节点判断为空时,结果得到了错误结果,因为每次不完全节点下一层遍历时均会结算,因此产生错误结果
- leetcode中国版出现了,不过我目前后悔将数据迁移到中文版