package main
import (
"container/list"
"fmt"
)
// Binary Tree
type BinaryTree struct {
Data interface{}
Left *BinaryTree
Right *BinaryTree
}
// Constructor
func NewBinaryTree(data interface{}) *BinaryTree {
return &BinaryTree{Data: data}
}
// 先序遍历-非递归
func (bt *BinaryTree) PreOrderNoRecursion() []interface{} {
t := bt
stack := list.New()
res := make([]interface{}, 0)
for t != nil || stack.Len() != 0 {
for t != nil {
res = append(res, t.Data)//visit
stack.PushBack(t)
t = t.Left
}
if stack.Len() != 0 {
v := stack.Back()
t = v.Value.(*BinaryTree)
t = t.Right
stack.Remove(v)
}
}
return res
}
// 中序遍历-非递归
func (bt *BinaryTree) InOrderNoRecursion() []interface{} {
t := bt
stack := list.New()
res := make([]interface{}, 0)
for t != nil || stack.Len() != 0 {
for t != nil {
stack.PushBack(t)
t = t.Left
}
if stack.Len() != 0 {
v := stack.Back()
t = v.Value.(*BinaryTree)
res = append(res, t.Data)//visit
t = t.Right
stack.Remove(v)
}
}
return res
}
// 后序遍历-非递归
func (bt *BinaryTree) PostOrderNoRecursion() []interface{} {
t := bt
stack := list.New()
res := make([]interface{}, 0)
var preVisited *BinaryTree
for t != nil || stack.Len() != 0 {
for t != nil {
stack.PushBack(t)
t = t.Left
}
v := stack.Back()
top := v.Value.(*BinaryTree)
if (top.Left == nil && top.Right == nil) || (top.Right == nil && preVisited == top.Left) || preVisited == top.Right{
res = append(res, top.Data)//visit
preVisited = top
stack.Remove(v)
}else {
t = top.Right
}
}
return res
}
func main() {
t := NewBinaryTree(1)
t.Left = NewBinaryTree(3)
t.Right = NewBinaryTree(6)
t.Left.Left = NewBinaryTree(4)
t.Left.Right = NewBinaryTree(5)
t.Left.Left.Left = NewBinaryTree(7)
fmt.Println(t.PreOrderNoRecursion())
fmt.Println(t.InOrderNoRecursion())
fmt.Println(t.PostOrderNoRecursion())
}
亿速云「云服务器」,即开即用、新一代英特尔至强铂金CPU、三副本存储NVMe SSD云盘,价格低至29元/月。点击查看>>
免责声明:本站发布的内容(图片、视频和文字)以原创、转载和分享为主,文章观点不代表本网站立场,如果涉及侵权请联系站长邮箱:is@yisu.com进行举报,并提供相关证据,一经查实,将立刻删除涉嫌侵权内容。