「这是我参与11月更文挑战的第 22 天,活动详情查看:2021最后一次更文挑战」
题目
给你二叉树的根节点 root
,返回它节点值的前序遍历。
示例 1:
输入:root = [1,null,2,3] 输出:[1,2,3]
示例 2:
输入:root = [] 输出:[]
示例 3:
输入:root = [1] 输出:[1]
示例 4:
输入:root = [1,2] 输出:[1,2]
示例 5:
输入:root = [1,null,2] 输出:[1,2]
提示:
树中节点数目在范围 [0, 100] 内 -100 <= Node.val <= 100
进阶:递归算法很简单,你可以通过迭代算法完成吗?
题解
提出问题
- 什么是二叉树的前序遍历?
分析
- 从根节点开始,先遍历左节点当左子树遍历完毕时,在遍历右子树
- 公式:前序遍历:根->左->右
递归 分析
- 定义
ans
用于存放遍历二叉树时的值 - 定义
traversal
方法用于实现二叉树的遍历 traversal
方法首先对root
进行非空判断当root
为空时直接返回null
- 否则套用公式
- 首先
push
根节点的值 - 如何递归调用
traversal(root.left,ans)
左子树已经traversal(root.right,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) {
let ans = []
traversal(root,ans)
return ans
};
let traversal = function(root,ans){
if(!root) return null
ans.push(root.val)
traversal(root.left,ans)
traversal(root.right,ans)
}
迭代 分析
- 除了使用递归的方式,我们还可以使用迭代的方式
- 定义
stack
栈,用于存历二叉树当前节点 - 定义
ans
用于存放遍历二叉树时的值 - 首先对
root
进行非空判断当root
不为空时将,当前根节点入栈push
- 通过
while
循环进行栈的遍历,直到stack
为空时结束 - 由于前序二叉树的遍历公式为前序遍历:根->左->右,以及此时用
stack
栈(先进后出)进行存储所有我们应当先把右子树push
入栈,再将左子树push
入栈 - 通过不断的
while
循环,执行上述操作 - 最终就实现二叉树的前序遍历
迭代 代码实现
/**
* 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 stack = []
let ans = []
if(root) stack.push(root)
while(stack.length > 0){
const curNode = stack.pop()
ans.push(curNode.val)
if(curNode.right !== null ){
stack.push(curNode.right)
}
if(curNode.left !== null ){
stack.push(curNode.left)
}
}
return ans
};