LeetCode-144 二叉树的前序遍历 JavaScript题解

思路

递归方法:
深度优先搜索

  • 先访问根结点
  • 根结点存在子节点
    • 访问左节点
    • 访问右节点

非递归方法:
以显示栈的方式模仿递归。

  • 初始化栈,并将根节点入栈;
  • 当栈不为空时:
    • 弹出栈顶元素 n,并将值添加到结果中
    • 如果 n 的右节点存在,入栈
    • 如果 n 的左节点存在,入栈
  • 输出结果数组

题解

方法一(非递归)
var preorderTraversal = function(root) {
  const stack = []
  const res = []
  // 1.若存在根结点,入栈(出栈)
  if (root) {stack.push(root)}
  // 3.若根结点有 左节点 或 右节点,循环
  while (stack.length) {
    // 1.1 根结点出栈,值添加到结果数组中存放
    const n = stack.pop()
    res.push(n.val)
    // 2.若存在右节点,左节点,入栈。推出栈顶,即,作为 根结点 出栈
    if(n.right) stack.push(n.right)
    if(n.left) stack.push(n.left)
  }
  return res
};
方法二 (递归)
var preorderTraversal = function(root, r = []) {
    if (!root) return []
    r.push(root.val)
    r = r.concat(preorderTraversal(root.left))
    r = r.concat(preorderTraversal(root.right))
    return r
};
方法三 (递归)

一行代码。现阶段最优解了

var preorderTraversal = function(root) {
  return root ? [root.val,...preorderTraversal(root.left), ...preorderTraversal(root.right)] : []
};

思考与改进

方法一

if (root) {stack.push(root)}

可写作
root && stack.push(root)

if(n.right) stack.push(n.right)

可写作
n.right && stack.push(n.right)

方法三

…是JS中扩展运算符。ES6中新增。

可以在函数调用/数组构造时,将数组表达式或者string在语法层面展开;还可以在构造字面量对象时将对象表达式按照key-value的方式展开。

作用约等于脱去括号。不管大括号([])、花括号({})。

...preorderTraversal(root.left)

在此处,可以理解为
[root.val].concat(preorderTraversal(root.left)

后续会再总结一篇关于(…)扩展运算符的文章。

  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 1
    评论
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值