LeetCode145、 二叉树的后序遍历

LeetCode 系列记录我学习算法的过程。

携手创作，共同成长！这是我参与「掘金日新计划 · 8 月更文挑战」的第24天，点击查看活动详情

题目

给你二叉树的根节点 root ，返回它节点值的 后序遍历。

示例：

输入： root = [1,null,2,3]
输出： [3,2,1]

输入： root = []
输出： []

提示：

• 树中节点数目在范围 [0, 100] 内
• -100 <= Node.val <= 100

进阶： 递归算法很简单，你可以通过迭代算法完成吗？

思路

二叉树常见的三种遍历方式分别是先序、中序和后序遍历，其中后序遍历的特点就是先遍历左子树，然后再遍历右子树树，最后遍历根结点

例如上图，后序遍历得到的结果是： 7 8 6 4 2 5 3 1

所以它也可以用先序遍历类似的递归的方式来解决：

• 首先判断根结点是否存在，不存在则直接返回空数组
• 定义一个空数组 arr 用于存储结果
• 判断左子树是否存在，存在则递归左子树，将结果赋值给 arr
• 然后判断右子树是否存在，存在则递归右子树，将结果与 arr 合并
• 最后将根结点的值存入，然后返回 arr 即可

代码实现

/**
 * 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) {
    // 如果当前根结点不存在，返回空数组
    if (!root) return []
    // 定义空数组来存储结果
    let arr = []
    // 如果左子树存在，则先递归左子树，将结果赋值给 arr
    if (root.left) {
        arr = postorderTraversal(root.left)
    }
    // 如果右子树存在，则递归右子树，将结果与 arr 合并
    if (root.right) {
        arr = [...arr, ...postorderTraversal(root.right)]
    }
    
    // 最后存入根结点的值
    arr = [...arr, root.val]
    // 返回结果
    return arr
};

优化

题目还给了一个进阶要求，就是不要用递归的方式，用迭代的方式来实现，其实和先序遍历的迭代方式差不多：

var postorderTraversal = function(root, res = []) {
    if (!root) return res;
    const stack = [root];
    let cur = null;
    do {
        cur = stack.pop();
        res.push(cur.val);
        cur.left && stack.push(cur.left);
        cur.right && stack.push(cur.right);
    } while(stack.length);
    return res.reverse();
};