golang 实现二叉树遍历

最新推荐文章于 2024-03-07 23:28:43 发布

lucky404

最新推荐文章于 2024-03-07 23:28:43 发布

阅读量2.2k

点赞数

分类专栏： go 文章标签： golang 二叉树 golang二叉树遍历

本文链接：https://blog.csdn.net/lucky404/article/details/92440857

版权

go 专栏收录该内容

26 篇文章 1 订阅

订阅专栏

前序遍历

 前序遍历的顺序是  根 -----> 左子树 -----> 右子树

中序遍历

中序遍历的顺序是 左子树 -----> 根 ------> 右子树

后序遍历

后序遍历的顺序是 左子树 -----> 右子树 -----> 根

package main

import "fmt"

// 二叉树的数据结构
type TreeNode struct {
	Data int
	Left *TreeNode
	Right *TreeNode
}

// 二叉树的实现
type Tree struct {
	root *TreeNode
}
// 添加数据
func (self *Tree) Add(data int) {
	var queue  []*TreeNode
	newNode := &TreeNode{Data:data}
	if self.root == nil {
		self.root = newNode
	}else {
		queue = append(queue, self.root)
		for len(queue) != 0 {
			cur := queue[0]
			queue = append(queue[:0], queue[0+1:]...)
			// 往右树添加
			if data > cur.Data {
				if cur.Right == nil {
					cur.Right = newNode
				} else {
					queue = append(queue, cur.Right)
				}
			// 往左数添加
			} else {
				if cur.Left == nil {
					cur.Left = newNode
				} else {
					queue = append(queue, cur.Left)
				}
			}
		}
	}
}
// 前序遍历 根 ---> 左 --->右
func (self *Tree )preorderTraverse(node *TreeNode)  {
	if node == nil {
		return
	} else {
		fmt.Print(node.Data, " ")
		self.preorderTraverse(node.Left)
		self.preorderTraverse(node.Right)
	}
	
}
// 中序遍历 左 ---> 根 --->右
func (self *Tree) inorderTraverse(node *TreeNode)  {
	if node == nil {
		return
	} else {
		self.inorderTraverse(node.Left)
		fmt.Print(node.Data, " ")
		self.inorderTraverse(node.Right)

	}
}
// 后序遍历 左 ----> 右 ---> 根
func (self *Tree) postTraverse(node *TreeNode)  {
	if node == nil {
		return
	} else {
		self.postTraverse(node.Left)
		self.postTraverse(node.Right)
		fmt.Print(node.Data, " ")
	}
}

func main()  {
	tree := &Tree{}
	tree.Add(50)
	tree.Add(45)
	tree.Add(40)
	tree.Add(48)
	tree.Add(51)
	tree.Add(61)
	tree.Add(71)

	fmt.Println("前序遍历")
	tree.preorderTraverse(tree.root)
	fmt.Println("")
	fmt.Println("中序遍历")
	tree.inorderTraverse(tree.root)
	fmt.Println("")
	fmt.Println("后续遍历")
	tree.postTraverse(tree.root)
}