1. 题目描述
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
来源:力扣(LeetCode)
链接:https://leetcode-cn.com/problems/binary-tree-preorder-traversal
著作权归领扣网络所有。商业转载请联系官方授权,非商业转载请注明出处。
2. 题解
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 preorderTraversal = function(root) {
let res = [];
function helper(root){
if(root){
res.push(root.val);
helper(root.left);
helper(root.right);
}
}
helper(root);
return res;
};
2、非递归
用栈来模拟递归的过程:
- 根节点入栈;
- 根节点出栈,并记录到结果数组;
- 遍历左子树,右子树,栈是先进后出,所以在入栈时按照先右子树后左子树的顺序入;
- 继续出栈…
直至栈空,遍历完成!
var preorderTraversal = function(root) {
let res = [];
let stack = []; //借助栈存储一些中间遍历结果
//初始状态:根节点不为空就入栈
if(root) stack.push(root);
while(stack.length > 0){
//出栈顶
let curr = stack.pop();
res.push(curr.val);
//右子节点先入栈(先进后出)
if(curr.right){
stack.push(curr.right);
}
//左子节点后入栈
if(curr.left){
stack.push(curr.left);
}
}
return res;
};
(因为结果存的是出栈顺序,所以入栈时先右子树后左子树。)