Go语言（container介绍）

今天，Mayuyu来介绍Go语言中一个重要的东西，叫做container。具体源码可以参考Go语言的源代码，如下

 

  路径：/usr/local/go/src/pkg/container

 

Github上的路径为：https://github.com/astaxie/gopkg/tree/master/container

 

container的结构如下

 

 

 

可以看出包含三部分：heap，list和ring。下面分别介绍

 

 

1. heap

 

   heap即为堆，是一种常用的数据结构，在源码里面，提供了接口，在实际使用时需要实现。

 

   下面是一个关于heap使用的实例。

 

   代码：

package main

import (
	"fmt"
	"container/heap"	
)

//heap提供了接口，需要自己实现如下方法
type Heap []int

//构造的是小顶堆，大顶堆只需要改一下下面的符号
func (h *Heap) Less(i, j int) bool {
	return (*h)[i] < (*h)[j]
}

func (h *Heap) Swap(i, j int) {
	(*h)[i], (*h)[j] = (*h)[j], (*h)[i]
}

func (h *Heap) Len() int {
	return len(*h)
}

func (h *Heap) Pop() interface{} {
	x := (*h)[h.Len() - 1]
	*h = (*h)[: h.Len() - 1]
	return x
}

func (h *Heap) Push(x interface{}) {
	*h = append(*h, x.(int))
}

func (h *Heap) Remove(idx int) interface{} {
	h.Swap(idx, h.Len() - 1)
	return h.Pop()
}

func main(){
	
	//创建一个heap
	h := &Heap{}
	
    heap.Init(h)
	//向heap中插入元素
	h.Push(5)
	h.Push(2)
	h.Push(1)
	h.Push(8)
	h.Push(4)
	h.Push(6)
	h.Push(2)

	//输出heap中的元素，相当于一个数组，原始数组
	fmt.Println(h)

	//这里必须要reheapify，建立好堆了
	heap.Init(h)

	//小顶堆对应的元素在数组中的位置
	fmt.Println(h)

	//移除下标为5的元素，下标从0开始
	h.Remove(5)

	//按照堆的形式输出
	for h.Len() > 0 {
		fmt.Printf("%d ", heap.Pop(h))
	}
	fmt.Println()
}

 

2. list

 

   list类型是双向链表，具体用法如下

 

   代码：

package main

import (
	"fmt"
	"container/list"	
)

func main(){

	//创建一个双向链表
	ls := list.New()

	//向双向链表中插入26个小写字母
	for i := 97; i < 123; i++ {
		ls.PushFront(i)     //PushFront()代表从头部插入，同样PushBack()代表从尾部插入
	}

	//遍历双向链表ls中的所有字母
	for it := ls.Front(); it != nil; it = it.Next() {
		fmt.Printf("%c ", it.Value)
	}
	fmt.Println()
}

 

3. ring

 

   ring是一个环形链表，具体用法可以参考如下代码

 

   代码：

package main

import (
	"fmt"
	"container/ring"	//闭环包引入，详见/usr/local/go/src/pkg/container/ring
)

func main(){
	//创建10个元素的闭环
	r := ring.New(10)

	//给闭环中的元素赋值
	for i := 1; i <= r.Len(); i++ {
		r.Value = i
		r = r.Next()
	}

    //循环打印闭环中的元素值
	r.Do(
		func(p interface{}){
			println(p)
	    })
	
	//获得当前元素之后的第5个元素
	r5 := r.Move(5)
	fmt.Println(r5)
	fmt.Println(r)

	//链接当前元素r与r5，相当于删除了r与r5之间的元素
	r1 := r.Link(r5)
	fmt.Println(r1)
	fmt.Println(r)
}