145. 二叉树的后序遍历

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
}