Pengjielee's blog

题目

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

示例 1：
1
 \
  2
 /
3 
输入：root = [1,null,2,3]
输出：[1,2,3]

示例 2：
输入：root = []
输出：[]

示例 3：
输入：root = [1]
输出：[1]

示例 4：
   1
  /
 2 
输入：root = [1,2]
输出：[1,2]

示例 5：
1 
  \
   2
输入：root = [1,null,2]
输出：[1,2]

提示：

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

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

来源：力扣（LeetCode）链接：https://leetcode-cn.com/problems/binary-tree-preorder-traversal 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路

前序遍历：根 -> 左 -> 右

JS实现

1、递归实现

/**
 * 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 helper = function (root, result) {
  if (root != null) {
    result.push(root.val);
    if (root.left != null) {
      helper(root.left, result);
    }
    if (root.right != null) {
      helper(root.right, result);
    }
  }
};

var preorderTraversal = function (root) {
  const result = [];
  if (root === null) {
    return result;
  }
  helper(root, result);
  return result;
};

2、迭代实现

var preorderTraversal = function (root) {
  const result = [];
  if (root === null) {
    return result;
  }

  // 栈的特点：先进后出
  const stack = [];

  // 先把根元素入栈
  stack.push(root);

  // 判断栈是否为空
  while (stack.length > 0) {
    // 出栈
    const node = stack.pop();
    // 保存节点值
    result.push(node.val);

    // 我们需要先取出左节点，所以把右节点先入栈
    if (node.right != null) {
      stack.push(node.right);
    }
    // 再入栈左节点
    if (node.left != null) {
      stack.push(node.left);
    }
  }
  return result;
};