给你二叉树的根节点 root
和一个表示目标和的整数 targetSum
。判断该树中是否存在 根节点到叶子节点 的路径,这条路径上所有节点值相加等于目标和 targetSum
。如果存在,返回 true
;否则,返回 false
。
叶子节点 是指没有子节点的节点。
示例 1:
输入:root = [5,4,8,11,null,13,4,7,2,null,null,null,1], targetSum = 22 输出:true 解释:等于目标和的根节点到叶节点路径如上图所示。
示例 2:
输入:root = [1,2,3], targetSum = 5 输出:false 解释:树中存在两条根节点到叶子节点的路径: (1 --> 2): 和为 3 (1 --> 3): 和为 4 不存在 sum = 5 的根节点到叶子节点的路径。
示例 3:
输入:root = [], targetSum = 0 输出:false 解释:由于树是空的,所以不存在根节点到叶子节点的路径。
思路:在此之前,我们做过找到二叉树的所有路径那道题,那再做这道题就很简单了,先拿到所有的路径,然后遍历看是否有条路径的总和是等于targetSum的。
代码(Python):
class Solution(object):
def hasPathSum(self, root, targetSum):
result = []
path = []
if not root:
return False
def backtracking(node,path):
path.append(node.val)
if not node.left and not node.right:
result.append(path[:])
return
ways = []
if node.left:
ways.append(node.left)
if node.right:
ways.append(node.right)
for way in ways:
backtracking(way,path)
path.pop()
backtracking(root,path)
for i in range(len(result)): #找是否有条路径的总和等于targetSum
if sum(result[i]) == targetSum:
return True
return False