一、题目内容 简单
给你二叉树的根节点 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
二、解题思路
我们使用前序遍历,用栈stack
来模拟,那所到路径的总和我们也用栈path
的数据结构。
我们要让 path
中的值,与 stack
的节点一一对应。
三、具体代码
/**
* @param {TreeNode} root
* @param {number} targetSum
* @return {boolean}
*/
var hasPathSum = function (root, targetSum) {
if (!root) return false
const stack = [root]
const path = [root.val]
while (stack.length) {
const node = stack.shift()
const sum = path.shift()
if (sum === targetSum && !node.left && !node.right) return true
if (node.right) {
stack.push(node.right)
path.push(sum + node.right.val)
}
if (node.left) {
stack.push(node.left)
path.push(sum + node.left.val)
}
}
return false
};
四、其他解法
/**
* @param {TreeNode} root
* @param {number} targetSum
* @return {boolean}
*/
var hasPathSum = function (root, targetSum) {
if (!root) return false
const backTracking = (node, sum) => {
if (sum === targetSum && !node.left && !node.right) return true
if (node.left) {
const result = backTracking(node.left, sum + node.left.val)
if (result) return true
}
if (node.right) {
const result = backTracking(node.right, sum + node.right.val)
if (result) return true
}
}
return backTracking(root, root.val) || false
};