golang實現二叉搜尋樹

這是一個建立於 的文章，其中的資訊可能已經有所發展或是發生改變。

關於什麼是二叉搜尋樹，不清楚的同學可以去看我寫的這個資料結構與演算法的網站

資料結構

首先我們定義需要的資料結構。注意，TreeNode的左右節點都是*TreeNode type的，而樹只有一個Root資料域，為*TreeNode type

type TreeNode struct {  Value int  Left *TreeNode  Right *TreeNode}type BinarySearchTree struct {  Root *TreeNode}

Insert

向二叉搜尋樹中插入元素，首先要找到插入的位置，然後再插入。這裡注意我們的實現方法為給TreeNode和BinarySearchTree這兩個type添加方法。需要注意給type添加方法的方式，同時還要注意，如果你要改變方法調用者，則需要使用指標

func (tree BinarySearchTree) Insert (v int) {  tree.Root.Insert(v)}func (node *TreeNode) Insert (v int){  if v < node.Value {    if node.Left != nil{      node.Left.Insert(v)    }else{      node.Left = &TreeNode{v, nil, nil}    }  }else {    if node.Right != nil{      node.Right.Insert(v)    }else{      node.Right = &TreeNode{v, nil, nil}    }  }}

遍曆

樹的遍曆有前序，後序，中序等等。這裡以中序為例。注意slice與slice指標的不同

func (tree BinarySearchTree) InOrder() []int{  var res []int  tree.Root.InOrder(&res)  return res}func (node *TreeNode) InOrder(result *[]int) {  if node.Left != nil{    node.Left.InOrder(result)  }  *result = append(*result, node.Value)  if node.Right != nil{    node.Right.InOrder(result)  }}

最大最小值

func (tree BinarySearchTree) FindMin() int {  node := tree.Root  for {    if node.Left != nil {      node = node.Left    }else{      return node.Value    }  }}func (tree BinarySearchTree) FindMax() int {  node := tree.Root  for {    if node.Right != nil {      node = node.Right    }else{      return node.Value    }  }}

尋找

func (tree BinarySearchTree) Contains(v int) bool {  node := tree.Root  for {    if node == nil{      return false    }else if (node.Value == v) {      return true    }else if (node.Value > v){      node = node.Left    }else{      node = node.Right    }  }}