LeetCode 145. 二叉树的后序遍历
题目描述
思路分析
后序遍历的顺序为:左子树 → 右子树 → 根节点。
参考代码
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
// 前序遍历
func preorderTraversal(root *TreeNode) []int {
var res []int
var dfs func(node *TreeNode)
dfs = func(node *TreeNode) {
if node == nil {
return
}
res = append(res, node.Val) // 根
dfs(node.Left) // 左
dfs(node.Right) // 右
}
dfs(root)
return res
}
// 中序遍历
func inorderTraversal(root *TreeNode) []int {
var res []int
var dfs func(node *TreeNode)
dfs = func(node *TreeNode) {
if node == nil {
return
}
dfs(node.Left) // 左
res = append(res, node.Val) // 根
dfs(node.Right) // 右
}
dfs(root)
return res
}
// 后序遍历
func postorderTraversal(root *TreeNode) []int {
var res []int
var dfs func(node *TreeNode)
dfs = func(node *TreeNode) {
if node == nil {
return
}
dfs(node.Left) // 左
dfs(node.Right) // 右
res = append(res, node.Val) // 根
}
dfs(root)
return res
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
// 后序遍历
func postorderTraversal(root *TreeNode) []int {
if root == nil {
return nil
}
var res []int
var stack []*TreeNode
stack = append(stack, root)
for len(stack) > 0 {
node := stack[len(stack)-1]
stack = stack[:len(stack)-1]
res = append(res, node.Val)
// 注意:先压入左子节点,再压入右子节点
if node.Left != nil {
stack = append(stack, node.Left)
}
if node.Right != nil {
stack = append(stack, node.Right)
}
}
// 反转结果数组
for i, j := 0, len(res)-1; i < j; i, j = i+1, j-1 {
res[i], res[j] = res[j], res[i]
}
return res
}
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
func postorderTraversal(root *TreeNode) []int {
if root == nil {
return []int{}
}
var stack []*TreeNode
var res []int
stack = append(stack, root)
for len(stack) > 0 {
cur := stack[len(stack)-1]
stack = stack[:len(stack)-1]
res = append([]int{cur.Val}, res...) // 逆序插入
if cur.Left != nil {
stack = append(stack, cur.Left)
}
if cur.Right != nil {
stack = append(stack, cur.Right)
}
}
return res
}
- 时间复杂度:O (n),其中 n 是二叉树的节点数。
- 空间复杂度:O (h),其中 h 是二叉树的高度。
1
write your code here
CC BY-NC-SA 4.0
许可协议,转载请注明出处!
本博客所有文章除特别声明外,均采用