用JavaScript遍历二叉树：后序遍历指南

前端

2023-12-28 11:41:43

后序遍历：深入浅出，掌握二叉树遍历技巧

什么是后序遍历？

在计算机科学领域，二叉树是一种广泛应用的数据结构，由一系列相互连接的节点组成，每个节点至多有两个子节点。后序遍历是一种访问二叉树中每个节点的方法，它以一种独特的顺序访问子节点和根节点：

左子节点： 首先访问左子节点。
右子节点： 然后访问右子节点。
根节点： 最后访问根节点。

例如，对于二叉树 [1, null, 2, 3]，后序遍历的结果为 [3, 2, 1]。

用 JavaScript 实现后序遍历

我们可以使用递归算法来实现后序遍历，该算法将问题分解为更小的子问题：

递归实现：

function postorderTraversal(root) {
  if (root === null) {
    return [];
  }

  const leftTraversal = postorderTraversal(root.left);
  const rightTraversal = postorderTraversal(root.right);
  return [...leftTraversal, ...rightTraversal, root.val];
}

优化：使用栈

为了优化后序遍历，我们可以使用栈数据结构，从而将时间复杂度从 O(n^2) 降低到 O(n)。

优化后的代码：

function postorderTraversalWithStack(root) {
  const stack = [root];
  const result = [];

  while (stack.length > 0) {
    const node = stack[stack.length - 1];

    if (node.left === null && node.right === null) {
      stack.pop();
      result.push(node.val);
    } else {
      if (node.right !== null) {
        stack.push(node.right);
      }
      if (node.left !== null) {
        stack.push(node.left);
      }
    }
  }

  return result;
}