LeetCode 144:二叉树的前序遍历

145 阅读2分钟

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

二叉树的前序遍历

题目描述:

给你二叉树的根节点 root ,返回它节点值的 前序 **遍历。

示例1

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

示例2

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

示例3

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

示例4

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

解题思路 

首先了解下前序遍历:
二叉树的前序遍历:按照访问根节点——左子树——右子树的方式遍历这棵树

思路一: 递归

我们只要首先将 root 节点的值加入result,然后递归调用 preorderTraversalNode(root.left) 来遍历 root 节点的左子树,最后递归调用 preorderTraversalNode(root.right) 来遍历 root 节点的右子树即可,递归终止的条件为碰到空节点

实现代码如下:

var preorderTraversal = function (root) {
  var result = [];
  var preorderTraversalNode = function (node) {
    if (node) {
      // 先根节点
      result.push(node.val);
      // 左子树
      preorderTraversalNode(node.left);
      // 右子树
      preorderTraversalNode(node.right);
    }
  }
  preorderTraversalNode(root);
  return result;
};

时间复杂度: O(n),其中 n 是二叉树的节点数

空间复杂度: O(n)

思路二: 迭代

利用栈来记录遍历的过程:

  1. 首先 根入栈;

  2. 根节点出栈,将根节点值存入数组中;

  3. 然后遍历左子树、右子树;因为栈是先进后出,所以右子树先入栈;

  4. 继续出栈(左子树出栈)

  5. 依次循环出栈遍历入栈,直到栈为空

与递归的区别:
递归的时候隐式地维护了一个栈,而我们在迭代的时候需要显式地将这个栈模拟出来

实现代码如下:

var preorderTraversal = function (root) {
  var result = [], stack = [];
  if (root) stack.push(root);
  while (stack.length) {
    var curr = stack.pop();
    result.push(curr.val);

    if (curr.right !== null) {
      stack.push(curr.right)
    }
    if (curr.left !== null) {
      stack.push(curr.left)
    }
  }
  return result;
};

时间复杂度:O(n),其中 n 是二叉树的节点数

空间复杂度:O(n)

参考资料

144. 二叉树的前序遍历 - 力扣(LeetCode)