题解 | #二叉树根节点到叶子节点的所有路径和#

二叉树根节点到叶子节点的所有路径和

http://www.nowcoder.com/practice/185a87cd29eb42049132aed873273e83

非递归实现,利用手动建栈来代替函数堆栈的实现方式。 情况总共分为三种情况,

  1. 父节点含有左右子节点。
  2. 父节点只有其中一边子节点。
  3. 父节点没有子节点。

如果要使用DFS深度遍历优先的方式来完成,则需要关注三个重点,就是路径选择的方向与路径是否重复以及路径计算保存的问题。

根据问题1路径选择方向,引入direction变量来判断方向。 根据问题2路径是否重复,选择以左边方向为默认,根据父节点拥有的子节点情况来选择方向以及是否将父节点压栈保留记录。 根据问题3路径计算保存,选择以更改子节点权值来进行保存。

JavaScript代码实现如下:

const sumNumbers = (root) => {
  // write code here
  // DFS
  let sum = 0;
  let eachRouteVal = 0;
  let nodeStack = [];
  let currentNode = root;
  let parent = null;
  let child = null;
  let direction = true; //true means left, false means right
  if (!root) return 0;
  if (!root.left && !root.right) return root.val;
  while (currentNode || !queueIsEmpty(nodeStack)) {
    parent = currentNode;
    if (currentNode.left || currentNode.right) {
      // only push parent got two children into stack,because of two path 
      if (currentNode.left && currentNode.right) {
        if (direction) {
          nodeStack.push(currentNode);
        }
        child = direction ? currentNode.left : currentNode.right;
      } else{
        direction = currentNode.left ? true : false;
        child = direction ? currentNode.left : currentNode.right;
      }
      eachRouteVal = calculateRouteValue(parent, child);
      child.val = eachRouteVal;
      currentNode = child;
      direction = true; //start from left route by default
    } else {
      // solve the right path
      currentNode = nodeStack.pop();
      sum += eachRouteVal;
      eachRouteVal = 0;
      direction = false;
    }
  }
  return sum;
}
const queueIsEmpty = (queue) =>{
    return queue.length === 0;
}
const calculateRouteValue = (parent,child) => {
    return parseInt([String(parent.val),String(child.val)].join(''));
}
全部评论

相关推荐

整体时间线:2月末力扣从零开始。3月初刷题成瘾,中旬陆续开面开杀,被机试折磨,下旬纠结日常offer选择。4月入职淘天,从硬landing到上手业务快乐融入5月平静美好,顺利到我觉得直接转正是最佳选择,月底转暑期流程被hr直接挂,主管诱骗能转正,万幸蚂蚁暑期流程没拒掉,压哨发意向,手里也还有个腾讯offer兜底,毁约腾讯暑期到此结束。==============================一些感悟:永远保留后手,先拿了阿里国际日常,拿到网易伏羲offer之后才拒绝意向,中间难免要催hr尽量开在同一时间,后续等淘天oc的时候立马拒了网易意向。不会让手里超过2个offer,但是也不会在未确定的时候就拒掉到手的。在淘天的时候师兄主管都保证能转正别担心,甚至主管拉我进内部群一起团建,但是始终把腾讯offer抓在手里,也给了我撕破脸之后和主管谈判的底气。蚂蚁一面二面间隔一个半月,时不时反向保温一下面试官又没拒掉流程,真是我最明智的选择。==============================实习体验:研一在鹅厂AI Lab实习打杂纯快乐的,自己包装一下也是有产出的。遇到的所有人都很温和有礼貌,整体不卷年纪偏大,公司关怀好,不考虑城市的话应该会是第一选择。淘天业务组非常业务,技术不容易提升但是容易有产出,整体强度能承受分到的活也不多还挺核心的,师兄还是很nice的,往年转正待遇也挺好,小组整体年龄结构有中有小没老人,晋升空间不错。拒掉的offer里面,同花顺是做大模型部署加速的,给钱少太卷拒了;阿里国际是研究型实习生随便面的感觉面试官技术没有太懂;网易伏羲是llm+智能npc其实很有搞头,还是贪图大厂title拒了;腾讯这个最可惜,agent+游戏ai,而且在大部门实习过可以丝滑landing,腾讯招聘经常能看到校招社招广告,应该是团队扩张期,考虑到城市因素忍痛拒绝,释放一个hc给大家。==============================彩蛋:想看看牛u会做什么选择,感觉人生到了这个时间点,每个决策都会影响很大,已知和女友都是浙江人,她稳定杭州工作,计划后续杭州定居结婚。 #暑期实习# #腾讯# #阿里# #蚂蚁# #大模型# #淘天#
投递蚂蚁集团等公司10个岗位
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务