[路飞]_leetcode 144. 二叉树的前序遍历 JavaScript版

427 阅读2分钟

「这是我参与11月更文挑战的第 22 天,活动详情查看:2021最后一次更文挑战

题目来源:leetcode 227. 二叉树的前序遍历

题目

给你二叉树的根节点 root ,返回它节点值的前序遍历。

示例 1:

输入:root = [1,null,2,3] 输出:[1,2,3]

示例 2:

输入:root = [] 输出:[]

示例 3:

输入:root = [1] 输出:[1]

示例 4:

输入:root = [1,2] 输出:[1,2]

示例 5:

输入:root = [1,null,2] 输出:[1,2]  

提示:

树中节点数目在范围 [0, 100] 内 -100 <= Node.val <= 100  

进阶:递归算法很简单,你可以通过迭代算法完成吗?

题解

提出问题

  • 什么是二叉树的前序遍历?

分析

  • 从根节点开始,先遍历左节点当左子树遍历完毕时,在遍历右子树
  • 公式:前序遍历:根->左->右

递归 分析

  • 定义ans用于存放遍历二叉树时的值
  • 定义traversal方法用于实现二叉树的遍历
  • traversal方法首先对root进行非空判断当root为空时直接返回null
  • 否则套用公式
  • 首先push根节点的值
  • 如何递归调用traversal(root.left,ans)左子树已经traversal(root.right,ans)右子树
  • 最终就实现二叉树的前序遍历

递归 代码实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
 // 递归
var preorderTraversal = function(root) {
    let ans = []
    traversal(root,ans)
    return ans
};

let traversal = function(root,ans){
    if(!root) return null
    ans.push(root.val)
    traversal(root.left,ans)
    traversal(root.right,ans)
}

迭代 分析

  • 除了使用递归的方式,我们还可以使用迭代的方式
  • 定义stack栈,用于存历二叉树当前节点
  • 定义ans用于存放遍历二叉树时的值
  • 首先对root进行非空判断当root不为空时将,当前根节点入栈push
  • 通过while循环进行栈的遍历,直到stack为空时结束
  • 由于前序二叉树的遍历公式为前序遍历:根->左->右,以及此时用stack栈(先进后出)进行存储所有我们应当先把右子树push入栈,再将左子树push入栈
  • 通过不断的while循环,执行上述操作
  • 最终就实现二叉树的前序遍历

迭代 代码实现

/**
 * Definition for a binary tree node.
 * function TreeNode(val, left, right) {
 *     this.val = (val===undefined ? 0 : val)
 *     this.left = (left===undefined ? null : left)
 *     this.right = (right===undefined ? null : right)
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
 // 迭代
var preorderTraversal = function(root) {
    let stack = []
    let ans = []
    if(root)  stack.push(root)
    while(stack.length > 0){
        const curNode = stack.pop()
        ans.push(curNode.val)
        if(curNode.right !== null ){
            stack.push(curNode.right)
        }
        if(curNode.left !== null ){
            stack.push(curNode.left)
        }
    }
    return ans
};