LeetCode145. 二叉树的后序遍历Golang版

LeetCode145. 二叉树的后序遍历Golang版

1. 问题描述

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

示例:

输入: [1,null,2,3]
1

2
/
3

输出: [3,2,1]
进阶: 递归算法很简单，你可以通过迭代算法完成吗？

2. 思路

2.1. 递归

2.2. 迭代

前序遍历：中左右
后序遍历：中右左 -> 左右中

1. 从根节点访问，到右分支叶子节点，边收集结果，边进栈
2. 弹栈，访问左孩子
3. 反转数组

3. 代码

3.1. 递归代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func postorderTraversal(root *TreeNode) []int {
    var res []int
    postorder(root, &res)
    return res
}

func postorder(root *TreeNode, res *[]int) {
    if root == nil {
        return
    }

    postorder(root.Left, res)
    postorder(root.Right, res)
    *res = append(*res, root.Val)
}

3.2. 迭代代码

/**
 * Definition for a binary tree node.
 * type TreeNode struct {
 *     Val int
 *     Left *TreeNode
 *     Right *TreeNode
 * }
 */
func postorderTraversal(root *TreeNode) []int {
    var res []int
    stack := []*TreeNode{}
    node := root

    for node != nil || len(stack) != 0 {
        for node != nil {
            res = append(res, node.Val)
            stack = append(stack, node)
            node = node.Right
        }
        node = stack[len(stack) - 1].Left
        stack = stack[:len(stack) - 1]
    }

    for i := 0; i < len(res) / 2; i++ {
        res[i], res[len(res) - i - 1] = res[len(res) - i - 1], res[i]
    }
    return res
}