题目
给你二叉树的根节点 root ,返回它节点值的 前序 遍历。
示例 1:
输入:root = [1,null,2,3]
输出:[1,2,3]
示例 2:
输入:root = []
输出:[]
示例 3:
输入:root = [1]
输出:[1]
思路
-
递归
- 递归函数参数:当前节点node,用于存放结果的数组ans
- 确定递归返回条件:当前节点为空,直接return
- 单层递归逻辑:前序遍历是’根左右’,先存放中间节点的值,然后递归左节点,再递归右节点
-
迭代
- 利用栈使用迭代的方法来完成前序遍历
- 首先将中间节点入栈
- 出栈一个节点,将该节点的右节点先入栈,然后再将左节点入栈。为什么右节点先入栈?这样出栈时才是左节点先出栈,右节点后出栈,也就是前序遍历的‘中左右’
- 栈为空时就结束
代码
- 递归
/**
* 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) {
function traversal(node , ans){
if(node === null) return
// 放入中间节点的值
ans.push(node.val)
// 递归左右节点
traversal(node.left , ans)
traversal(node.right , ans)
}
let ans = []
traversal(root , ans)
return ans
};
- 迭代
/**
* 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) {
if(!root) return []
let queue = []
let ans = []
queue.push(root)
while(queue.length){
const node = queue.pop()
ans.push(node.val)
if(node.right) queue.push(node.right)
if(node.left) queue.push(node.left)
}
return ans
};
关注我的专栏,每天更新三道leetcode题解,一起变强!