问题

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

示例 1:

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

示例 2:

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

示例 3:

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

提示:

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

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

方法一: 递归

前、中、后序遍历都是基于DFS,节点的访问顺序如上图所示,每个节点有三个不同的驻留阶段,即每个节点会被经过三次:

  • 在递归它的左子树之前。
  • 在递归完它的左子树之后,在递归它的右子树之前。
  • 在递归完它的右子树之后。
const inorderTraversal = (root) => {
    const res = [];
    const inorder = (root) => {
        if (root == null) {
            return;
        }
        inorder(root.left); // 先递归左子树
        res.push(root.val); // 将当前节点值推入res
        inorder(root.right); // 再递归右子树
    };
    inorder(root);
    return res;
};

方法二: 中序遍历的迭代实现

const inorderTraversal = (root) => {
  const res = [];
  const stack = [];
 
  while (root) {        // 能压栈的左子节点都压进来
    stack.push(root);
    root = root.left;
  }
  while (stack.length) {
    let node = stack.pop(); // 栈顶的节点出栈
    res.push(node.val);     // 在压入右子树之前,处理它的数值部分(因为中序遍历)
    node = node.right;      // 获取它的右子树
    while (node) {          // 右子树存在,执行while循环    
      stack.push(node);     // 压入当前root
      node = node.left;     // 不断压入左子节点
    }
  }
  return res;
};