[路飞]_leetcode 94. 二叉树的中序遍历 JavaScript版

「这是我参与11月更文挑战的第 24 天，活动详情查看：2021最后一次更文挑战」

题目来源：leetcode 94. 二叉树的中序遍历

给定一个二叉树的根节点 root ，返回它的 中序 遍历。

 

示例 1：

输入：root = [1,null,2,3] 输出：[1,3,2] 示例 2：

输入：root = [] 输出：[] 示例 3：

输入：root = [1] 输出：[1] 示例 4：

输入：root = [1,2] 输出：[2,1] 示例 5：

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

提示：

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

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

题解

提出问题

• 什么是二叉树的中序遍历？

分析

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

递归 分析

• 定义ans用于存放遍历二叉树时的值
• 定义traversal方法用于实现二叉树的遍历
• traversal方法首先对root进行非空判断当root为空时直接返回null
• 否则套用公式
• 首先递归调用traversal(root.left,ans)左子树
• 再push根节点的值
• 最后 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 inorderTraversal = function(root) {
    let ans = []
    traversal(root,ans)
    return ans
};

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

迭代 分析

• 除了使用递归的方式，我们还可以使用迭代的方式
• 定义stack栈，用于存历二叉树当前节点
• 定义ans用于存放遍历二叉树时的值
• 通过while循环进行栈的遍历，直到stack为空时结束或者root为null
• 首先通过while循环对root进行非空判断当root不为空时将，当前根节点入栈push，root复制为当前节点的左节点
• 通过不断的while循环，执行上述操作，先将左节点都入栈push
• 然后出栈pop出栈当前节点curNode，并将值push到数组ans中
• 在将root赋值成前节点curNode右节点，执行上述操作
• 最终就实现二叉树的中序遍历

迭代 代码实现

/**
 * 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 inorderTraversal = function(root) {
    let stack = []
    let ans = []
    while(stack.length > 0 || root){
        while(root) {
            stack.push(root)
            root = root.left
        }
        let curNode = stack.pop()
        ans.push(curNode.val)
        root = curNode.right
    }
    return ans
};