给你二叉树的根节点
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 解释:由于树是空的,所以不存在根节点到叶子节点的路径。提示:
- 树中节点的数目在范围
[0, 5000]
内-1000 <= Node.val <= 1000
-1000 <= targetSum <= 1000
思路:
- 递归法,较简单
- 迭代法 - BFS:判断当前节点值是否等于targetSum,如果相等则返回true,否则下探至当前节点的左右子节点,并不断累加之前父节点的值,用于后续比较。
时间复杂度:O(N),其中 N 是树的节点数,对每个节点访问一次。
空间复杂度: O(N),其中 N 是树的节点数。空间复杂度主要取决于队列的开销,队列中的元素个数最多不会超过树的节点数。
/**
* Definition for a binary tree node.
* type TreeNode struct {
* Val int
* Left *TreeNode
* Right *TreeNode
* }
*/
// 解法一 递归法 DFS:
// 利用减法
func hasPathSum(root *TreeNode, targetSum int) bool {
// 走到空了,还没找到累加和为targetSum的路径,那么此路径返回false
if root == nil {
return false
}
// 叶子节点
if root.Left == nil && root.Right == nil && root.Val == targetSum {
return true
}
l := hasPathSum(root.Left, targetSum - root.Val)
r := hasPathSum(root.Right, targetSum - root.Val)
return l || r
}
// 利用加法 需额外增加sum入参
func hasPathSum(root *TreeNode, targetSum int) bool {
var dfs func(root *TreeNode, sum int) bool
dfs = func(root *TreeNode, sum int) bool {
if root == nil {
// 走到空了,还没找到累加和为targetSum的路径,那么此路径返回false
return false
}
sum += root.Val
// 叶子节点
if root.Left == nil && root.Right == nil && sum == targetSum {
return true
}
return dfs(root.Left, sum) || dfs(root.Right, sum)
}
return dfs(root, 0)
}
// 解法二 迭代法 BFS(queue):判断当前节点值是否等于targetSum,并且当前左右子节点不断累加之前父节点的值,用于后续比较
func hasPathSum(root *TreeNode, targetSum int) bool {
if root == nil {
return false
}
queue := []*TreeNode{root}
for len(queue) > 0 {
root = queue[0] // top
queue = queue[1:] // pop
// 叶子节点
if root.Left == nil && root.Right == nil && root.Val == targetSum {
return true
}
if root.Left != nil {
queue = append(queue, root.Left)
root.Left.Val += root.Val
}
if root.Right != nil {
queue = append(queue, root.Right)
root.Right.Val += root.Val
}
}
return false
}
路径总和 系列题目: