leetcode 145. 二叉树的后序遍历 JavaScript版

「这是我参与11月更文挑战的第 23 天，活动详情查看：2021最后一次更文挑战」

题目来源：leetcode 145. 二叉树的后序遍历

给定一个二叉树，返回它的 后序 遍历。

示例:

输入: [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
};
复制代码