LeetCode探索(27):112-路径总和

166 阅读2分钟

「这是我参与2022首次更文挑战的第13天,活动详情查看:2022首次更文挑战」。

题目

给你二叉树的根节点 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

解答

方法一:递归

这是二叉树相关的一道题目,难度简单。我们可以使用递归的方法,每次比较路径中的节点的和与目标和targetSum是否相等。递归解法的代码看起来比较精炼、易懂。

var hasPathSum = function(root, targetSum) {
  if (!root) {
    return false
  }
  if (!root.left && !root.right) {
    return root.val === targetSum
  }
  return hasPathSum(root.left, targetSum - root.val) || hasPathSum(root.right, targetSum - root.val)
}
// 执行用时:76 ms, 在所有 JavaScript 提交中击败了82.23%的用户
// 内存消耗:44.2 MB, 在所有 JavaScript 提交中击败了6.10%的用户

复杂度分析

  • 时间复杂度:O(N),其中 N 是树的节点数。对每个节点访问一次。

  • 空间复杂度:O(H),其中 H 是树的高度。最坏情况下,树呈现链状,空间复杂度为 O(N)。

方法二:广度优先搜索

我们使用两个队列,分别存储将要遍历的节点,以及根节点到这些节点的路径和。

var hasPathSum = function(root, targetSum) {
  if(!root) return false
  const queNode = []
  const queVal = []
  queNode.push(root)
  queVal.push(root.val)
  while (queNode.length) {
    // 每次取出队列中的节点和节点值的和
    let now = queNode.shift()
    let temp = queVal.shift()
    // 比较
    if (!now.left && !now.right) {
      if (temp == targetSum) {
        return true
      }
      continue
    }
    if (now.left) {
      queNode.push(now.left)
      queVal.push(now.left.val + temp)
    }
    if (now.right) {
      queNode.push(now.right)
      queVal.push(now.right.val + temp)
    }
  }
  return false
}
// 执行用时:68 ms, 在所有 JavaScript 提交中击败了97.70%的用户
// 内存消耗:44.9 MB, 在所有 JavaScript 提交中击败了5.01%的用户

复杂度分析

  • 时间复杂度:O(N),其中 N 是树的节点数。对每个节点访问一次。
  • 空间复杂度:O(N),其中 N 是树的节点数。

参考