热门IT资讯网

golang二叉树前序,中序,后序非递归遍历算法

发表于:2024-11-26 作者:热门IT资讯网编辑
编辑最后更新 2024年11月26日,package mainimport ( "container/list" "fmt")// Binary Treetype BinaryTree struct { Data inter
package mainimport (   "container/list"   "fmt")// Binary Treetype BinaryTree struct {   Data  interface{}   Left  *BinaryTree   Right *BinaryTree}// Constructorfunc 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())}


0