144. 二叉树的前序遍历(JavaScript)

给定一个二叉树,返回它的 前序 遍历。

 示例:

输入: [1,null,2,3]  
   1
    \
     2
    /
   3 

输出: [1,2,3]

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


解法一:递归法

递归法很简单,按照根节点、左子树、右子树的顺序。JavaScript中数组的连接使用concat,且此函数并不改变数组本身。

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var preorderTraversal = function(root) {
  if (!root) return [];
  var result = [];
  result.push(root.val);
  result = result.concat(preorderTraversal(root.left));
  result = result.concat(preorderTraversal(root.right));
  return result;
};

解法二:迭代法

迭代法意在通过循环解决问题。建立一个栈,将根节点加入栈中,当栈不为空时:

1、取出栈顶元素 curr,访问curr

2、若curr右子节点不为空,则将curr右子节点加入栈

3、若curr左子节点不为空,则将curr左子节点加入栈

/**
 * Definition for a binary tree node.
 * function TreeNode(val) {
 *     this.val = val;
 *     this.left = this.right = null;
 * }
 */
/**
 * @param {TreeNode} root
 * @return {number[]}
 */
var preorderTraversal = function(root) {
  if (!root) return [];
  var result = [],
      stack = [root];
  while (stack.length !== 0) {
    var curr = stack.pop();
    if (curr.right) stack.push(curr.right);
    if (curr.left) stack.push(curr.left);
    result.push(curr.val);
  }
  return result;
};

猜你喜欢

转载自blog.csdn.net/romeo12334/article/details/81477219