-
Notifications
You must be signed in to change notification settings - Fork 8
/
main.go
83 lines (69 loc) · 1.41 KB
/
main.go
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
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
package main
import(
"fmt"
"container/list"
)
type TreeNode struct {
Val interface{}
Left *TreeNode
Right *TreeNode
}
/*
The binary tree in this example is shown as follow
1
/ \
2 3
/ \ /
4 5 6
/
7
post-order traversal: 4 7 5 2 6 3 1
*/
func (tn *TreeNode) PostorderTraversalByRecursion() {
if tn == nil {
return
}
tn.Left.PostorderTraversalByRecursion()
tn.Right.PostorderTraversalByRecursion()
fmt.Print(tn.Val, " ")
}
func (tn *TreeNode) PostorderTraversal() {
if tn == nil {
return
}
stack := list.New()
stack.PushBack(tn)
var pre *TreeNode
for stack.Len() > 0 {
s := stack.Back()
cur := s.Value.(*TreeNode)
switch {
case cur.Left != nil && cur.Left != pre && cur.Right != pre:
stack.PushBack(cur.Left)
case cur.Right != nil && cur.Right != pre:
stack.PushBack(cur.Right)
default:
fmt.Print(cur.Val, " ")
pre = cur
stack.Remove(s)
}
}
}
func main() {
nodes := []int{1, 2, 3, 4, 5, 6, -1, -1, -1, 7}
root := generateBTree(nodes, 0)
root.PostorderTraversalByRecursion()
fmt.Println()
root.PostorderTraversal()
fmt.Println()
}
func generateBTree(nodes []int, index int) *TreeNode {
var node TreeNode
if index < len(nodes) && nodes[index] != -1 {
node = TreeNode{Val: nodes[index]}
node.Left = generateBTree(nodes, index * 2 + 1)
node.Right = generateBTree(nodes, index * 2 + 2)
return &node
}
return nil
}