LeetCode94、二叉树的中序遍历

LeetCode 系列记录我学习算法的过程。

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第9天，点击查看活动详情

题目

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

示例：

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


输入： root = []
输出： []

提示

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

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

思路

这个题目给的示例非常简陋，不懂二叉树的人可能连题目的意思都看不懂

所谓的中序遍历，其实是二叉树遍历的其中一种方式，先遍历左子树，然后是根节点，最后遍历右子树，例如

这个示例按中序遍历输出就是：[4, 7, 6, 8, 2, 1, 3, 5]

对于这种题目一般都是用递归的方式来进行处理，二叉树上的每一个节点的处理方式都是一样的

先判断左子树是否存在，存在则先遍历左子树；然后将根节点值存储起来；然后判断右子树是否存在，存在则遍历右子树

通过这样递归调用即可把二叉树所有节点按中序遍历的方式输出

代码实现

/**
 * 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) {
    // res 存储遍历的节点值
    let res = []
    // 如果根节点为 null，直接返回空数组
    if (!root) return res
    // 定义递归函数
    function p(tree) {
        // 左子树存在则优先遍历左子树
        if (tree.left) p(tree.left)
        // 然后将当前节点值存入数组
        res.push(tree.val)
        // 再判断右子树是否存在，存在则遍历
        if (tree.right) p(tree.right)
    }
    // 调用递归函数
    p(root)
    // 返回结果
    return res
};

优化

这题还可以通过迭代的方式实现，思路和递归差不多

var inorderTraversal = function(root) {
    // res 存储结果
    const res = []
    // 栈
    const stk = []
    while (root || stk.length) {
        // 遍历左子树，存入栈
        while (root) {
            stk.push(root)
            root = root.left
        }
        // 出栈，存值
        root = stk.pop()
        res.push(root.val)
        // 指向右子树
        root = root.right
    }
    return res
}