145. 二叉树的后序遍历
145. 二叉树的后序遍历
给定一个二叉树,返回它的 后序 遍历。 示例:
输入: [1,null,2,3]
1
\
2
/
3
输出: [3,2,1]
1. 递归解决
func postorderTraversal(root *TreeNode) []int {
var result []int
var postororder func(root *TreeNode)
postororder = func(node *TreeNode) {
if node == nil {
return
}
// left -> right -> root
postororder(node.Left)
postororder(node.Right)
result = append(result, node.Val)
}
postororder(root)
return result
}
2. 迭代,节点标记法
在出入栈的时候,标记节点,具体为: 标记节点的状态,新节点为false,已使用(在这道题里是指将节点值追加到结果数组)的节点true。 如果遇到未标记的节点,则将其标记为true,然后将其自身、右节点、左节点依次入栈;注意到顺序与遍历次序正好相反。 如果遇到的节点标记为true,则使用该节点。 这个方法在前序、中序、后续遍历里的实现代码总体逻辑一致,只是入栈的顺序稍微调整即可
func postorderTraversal2(root *TreeNode) []int {
var result []int
stack := []*TreeNode{root}
marked := map[*TreeNode]bool{}
for len(stack) > 0 {
node := stack[len(stack)-1]
stack = stack[:len(stack)-1]
if node == nil {
continue
}
if marked[node] {
result = append(result, node.Val)
continue
}
marked[node] = true
stack = append(stack, node)
stack = append(stack, node.Right)
stack = append(stack, node.Left)
}
return result
}
3. morris 迭代实现
优秀的是空间复杂度 详见 Morris 迭代实现解法说明
var res []int
func postorderTraversalMorris(root *TreeNode) []int {
res = nil
dummy := &TreeNode{Left: root}
cur := dummy
var node *TreeNode
for cur != nil {
if cur.Left == nil {
cur = cur.Right
continue
}
// 找 cur 的前驱
node = cur.Left
for node.Right != nil && node.Right != cur {
node = node.Right
}
if node.Right == nil { // 还没线索化,建立线索
node.Right = cur
cur = cur.Left
} else { // 已经线索化,访问节点并删除线索以恢复树的结构
node.Right = nil
visitPath(cur.Left)
cur = cur.Right
}
}
dummy.Left = nil
return res
}
func visitPath(node *TreeNode) {
end := reversePath(node)
for p := end; p != nil; p = p.Right {
res = append(res, p.Val)
}
_ = reversePath(end)
}
func reversePath(node *TreeNode) *TreeNode {
var prev *TreeNode
for node != nil {
prev, node, node.Right = node, node.Right, prev
}
return prev
}