Golang实现二叉搜索树

丶老邪 · · 1167 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

现在只实现了很简单的增加、查找、遍历。后面有空会完善。

package main

import "fmt"

type BNode struct {
	data  int
	left  *BNode
	right *BNode
}

type Tree struct {
	FirstNode   *BNode
}

func (t *Tree) PrintData (node *BNode) {
	if node.left != nil {
		t.PrintData(node.left)
	}
	fmt.Println(node.data)
	if node.right != nil {
		t.PrintData(node.right)
	}


	return
}

// 返回node
func (t *Tree) Search(data int, bnode *BNode) *BNode {
	if bnode == nil {
		return nil
	}
	if data > bnode.data {
		return t.Search(data, bnode.right)
	} else if data < bnode.data {
		return t.Search(data, bnode.left)
	} else if data == bnode.data {
		return bnode
	}
	return nil
}

func (t *Tree) Insert(data int, currentNode *BNode) {
	newNode := &BNode{
		data,
		nil,
		nil,
	}
	if t.FirstNode == nil {
		t.FirstNode = newNode
		return
	}
	if t.Search(data, t.FirstNode) != nil {
		panic("该数据已经存在")
		return
	}

	if data > currentNode.data {
		if currentNode.right == nil {
			currentNode.right = newNode
		} else {
			t.Insert(data,currentNode.right)
		}
	} else if data < currentNode.data {
		if currentNode.left == nil {
			currentNode.left = newNode
		} else {
			t.Insert(data, currentNode.left)
		}
	}

	return
}

func main() {
	nodes := []int{4,1,7,15,10}
	tree := Tree{}
	for _, data := range nodes {
		tree.Insert(data, tree.FirstNode)
	}

	tree.PrintData(tree.FirstNode)
	fmt.Print(tree.Search(10, tree.FirstNode))
}

 


有疑问加站长微信联系(非本文作者)

本文来自:开源中国博客

感谢作者:丶老邪

查看原文:Golang实现二叉搜索树

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

1167 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传