给定一个二叉树,返回它的 前序 遍历。
示例:
输入: [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;
};