每日算法-路径总和(树-简单)

296 阅读2分钟

题目

给定一个二叉树和一个目标和,判断该树中是否存在根节点到叶子节点的路径,这条路径上所有节点值相加等于目标和。 说明: 叶子节点是指没有子节点的节点。 示例:

给定如下二叉树,以及目标和 sum = 22,
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

返回 true, 因为存在目标和为 22 的根节点到叶子节点的路径 5->4->11->2。

方法一:dns-深度优先遍历

sum —— 从根节点到叶子节点的路径上的节点值相加的目标和 递归。转为判断:左、右子树中能否找出和为 sum - root.val 的路径 每遍历一个节点,sum 就减去当前节点值,当遍历到叶子节点时,已经没有子节点了,如果 sum - 当前叶子节点值 == 0 ,就是找到了从根节点到叶子节点的和为 sum 的路径

const hasPathSum = (root, sum) => {
  if (root == nullreturn false;                // 遍历到null节点
  if (root.left == null && root.right == null) { // 遍历到叶子节点
    return sum - root.val == 0;                  // 如果满足这个就返回true
  }
  return hasPathSum(root.left, sum - root.val) ||
    hasPathSum(root.right, sum - root.val);      // 大问题转成两个子树的问题
}

方法二:bfs-广度优先遍历

通过数组保存广度遍历对应的结果,依次出队列;

var hasPathSum = function (root, sum) {
  if (root === null) return false;
  let que = [
    root
  ]
  let quePath = [
    root.val
  ]
  while(que) {
    let node = que.shift();
    let temp = quePath.shift()
    if (node.left === null && node.right === null && path === sum) return true;
    if (node.left) {
      que.push(node.left)
      quePath.push(node.left.val+temp)
    }
    if (node.right) {
      que.push(node.right)
      quePath.push(node.right.val+temp)
    }
  }
  return false;
}