前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >一天一大 leet(路径总和)难度:简单-Day20200707

一天一大 leet(路径总和)难度:简单-Day20200707

作者头像
前端小书童
发布2020-09-24 11:41:03
1420
发布2020-09-24 11:41:03
举报
文章被收录于专栏:前端小书童

题目:

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

说明

叶子节点是指没有子节点的节点。

示例

给定如下二叉树,以及目标和 sum = 22,

代码语言:javascript
复制
              5
             / \
            4   8
           /   / \
          11  13  4
         /  \      \
        7    2      1

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

抛砖引玉

  • 使用递归遍历二叉树
  • 递归的逻辑:
    • 参数上一个节点的左节点或者右节点,减去上一个节点的和
    • 遇到叶子节点下一个节点终止
    • sum 减去递归中遇到的节点,返回 sum 与最后的节点差值是否为 0
代码语言:javascript
复制
/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} sum
 * @return {boolean}
 */
var hasPathSum = function (root, sum) {
  // 遍历到叶子节点下一个节点-终止递归
  if (root === null) return false

  // 遍历到叶子节点 判断剩余节点是否与和差值为0
  if (root.left === null && root.right === null) {
    return sum - root.val === 0
  }

  // sum一次减去该节点左节点、右节点 是否等于最后叶子节点
  let left = hasPathSum(root.left, sum - root.val),
    right = hasPathSum(root.right, sum - root.val)

  // 存在一侧满足就满足
  return left || right
}

官方答案

广度优先搜索
  • 题目要求根节点到叶子节点所以如果每遍历一层节点时, 我们找到该层所以的节点,理论上任何一个节点我们都可以与上一节点累加去校验逻辑
  • 现在的问题是,如果只单纯的存储每一层所以的节点,就都是了每个节点的父级节点,不知道了与谁相加
  • 那么尝试在存入节点时就存入处理后的值(借助递归方法中使用的依次减掉已经遇到的节点)

代码实现

  • 声明一个数组 stack 记录每层收集的节点,放入树形数据-第一层(栈-先入先出)
  • 声明一个数组 sumStack 记录每层目标值与累加结果的差值
  • 遍历 stack,只要这一层不全为 null,就一直遍历直到叶子节点终止
  • 取出上一次节点,存入 stack 中节点的左右节点存入做下一层节点
  • 判断 sumStack 存储的差值是否为 0,及为 0 是存入的节点是不是叶子节点

注意

stack 与 sumStack 的数据是一一对应的,遍历中会遍历所有节点及根节点到任意节点的差值

代码语言:javascript
复制
/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @param {number} sum
 * @return {boolean}
 */
var hasPathSum = function (root, sum) {
  if (!root) return false
  let stack = [root],
    sumStack = [sum - root.val]

  while (stack.length) {
    let node = stack.pop(),
      curSum = sumStack.pop()
    if (curSum === 0 && !node.left && !node.right) {
      return true
    }
    if (node.left) {
      stack.push(node.left)
      sumStack.push(curSum - node.left.val)
    }
    if (node.right) {
      stack.push(node.right)
      sumStack.push(curSum - node.right.val)
    }
  }
  return false
}
本文参与 腾讯云自媒体同步曝光计划,分享自微信公众号。
原始发表:2020-07-07,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 前端小书童 微信公众号,前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 说明
  • 示例
  • 抛砖引玉
  • 官方答案
    • 广度优先搜索
    相关产品与服务
    对象存储
    对象存储(Cloud Object Storage,COS)是由腾讯云推出的无目录层次结构、无数据格式限制,可容纳海量数据且支持 HTTP/HTTPS 协议访问的分布式存储服务。腾讯云 COS 的存储桶空间无容量上限,无需分区管理,适用于 CDN 数据分发、数据万象处理或大数据计算与分析的数据湖等多种场景。
    领券
    问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档