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