LeetCode探索(22):94-二叉树的中序遍历

144 阅读2分钟

「这是我参与2022首次更文挑战的第8天,活动详情查看:2022首次更文挑战」。

题目

给定一个二叉树的根节点 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

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

解答

方法一:递归

/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var inorderTraversal = function(root, res = []) {
  if (!root) return res
  inorderTraversal(root.left, res)
  res.push(root.val)
  inorderTraversal(root.right, res)
  return res
}

复杂度分析

  • 时间复杂度:O(n),其中 n 是二叉树的结点个数。二叉树的遍历中每个节点会被访问一次且只会被访问一次。
  • 空间复杂度:O(n),空间复杂度取决于递归的栈深度。

方法二:迭代

略,可以参考 144-二叉树的前序遍历,二者是类似的。该方法的时间复杂度是O(n),空间复杂度是O(n)

方法三:Morris 中序遍历

用递归和迭代的方式都使用了辅助空间,而Morris 中序遍历没有使用任何辅助空间。但是,缺点是改变了整个树的结构,把一棵二叉树改成了链表结构。

首先,我们找到 root 的左节点的最右子树,将当前 root 节点连带右子树全部挂到左节点的最右子树下面。

依此类推,直到新的二叉树没有左子树了,则直接访问右子树,每次把节点的值存入数组res,最后返回该数组即可。

var inorderTraversal = function(root) {
  const res = [];
  let predecessor = null;
  while (root) {
    if (root.left) { // 一、如果有左子树
      // 1. 找到 root 的左节点的最右子树
      predecessor = root.left;
      while (predecessor.right && predecessor.right !== root) {
        predecessor = predecessor.right;
      }
      // 2. 将当前 root 节点连带右子树全部挂到左节点的最右子树下面
      if (!predecessor.right) {
        predecessor.right = root;
        root = root.left;
      } else { // 3. 说明左子树已经访问完了,我们需要断开链接
        res.push(root.val);
        predecessor.right = null;
        root = root.right;
      }
    } else { // 二、如果没有左子树,则直接访问右子树
      res.push(root.val); // 把节点的值存入数组res
      root = root.right;
    }
  }
  return res;
};

复杂度分析

  • 时间复杂度:O(n),其中 n 为二叉搜索树的节点个数。Morris 遍历中每个节点会被访问两次,因此总时间复杂度为 O(2n) = O(n)。
  • 空间复杂度:O(1)。

参考