（五）golang开箱即用container heap、list

栈

package main
 
import (
    "container/heap"
    "fmt"
)
 
type IntHeap []int
 
//我们自定义一个堆需要实现5个接口
//Len(),Less(),Swap()这是继承自sort.Interface
//Push()和Pop()是堆自已的接口
 
//返回长度
func (h *IntHeap) Len() int {
    return len(*h);
}
 
//比较大小(实现最小堆)
func (h *IntHeap) Less(i, j int) bool {
    return (*h)[i] < (*h)[j];
}
 
//交换值
func (h *IntHeap) Swap(i, j int) {
    (*h)[i], (*h)[j] = (*h)[j], (*h)[i];
}
 
//压入数据
func (h *IntHeap) Push(x interface{}) {
    //将数据追加到h中
    *h = append(*h, x.(int))
}
 
//弹出数据
func (h *IntHeap) Pop() interface{} {
    old := *h;
    n := len(old);
    x := old[n-1];
    //让h指向新的slice
    *h = old[0: n-1];
    //返回最后一个元素
    return x;
}
 
//打印堆
func (h *IntHeap) PrintHeap() {
    //元素的索引号
    i := 0
    //层级的元素个数
    levelCount := 1
    for i+1 <= h.Len() {
        fmt.Println((*h)[i: i+levelCount])
        i += levelCount
        if (i + levelCount*2) <= h.Len() {
            levelCount *= 2
        } else {
            levelCount = h.Len() - i
        }
    }
}
 
func main() {
    a := IntHeap{6, 2, 3, 1, 5, 4};
    //初始化堆
    heap.Init(&a);
    a.PrintHeap();
    //弹出数据，保证每次操作都是规范的堆结构
    fmt.Println(heap.Pop(&a));
    a.PrintHeap();
    fmt.Println(heap.Pop(&a));
    a.PrintHeap();
    heap.Push(&a, 0);
    heap.Push(&a, 8);
    a.PrintHeap();
}

list链表

package main;
 
import (
    "container/list"
    "fmt"
)
 
func printList(l *list.List) {
    for e := l.Front(); e != nil; e = e.Next() {
        fmt.Print(e.Value, " ");
    }
    fmt.Println();
}
 
func main() {
    //创建一个链表
    l := list.New();
 
    //链表最后插入元素
    a1 := l.PushBack(1);
    b2 := l.PushBack(2);
 
    //链表头部插入元素
    l.PushFront(3);
    l.PushFront(4);
 
    printList(l);
 
    //取第一个元素
    f := l.Front();
    fmt.Println(f.Value);
 
    //取最后一个元素
    b := l.Back();
    fmt.Println(b.Value);
 
    //获取链表长度
    fmt.Println(l.Len());
 
    //在某元素之后插入
    l.InsertAfter(66, a1);
 
    //在某元素之前插入
    l.InsertBefore(88, a1);
 
    printList(l);
 
    l2 := list.New();
    l2.PushBack(11);
    l2.PushBack(22);
    //链表最后插入新链表
    l.PushBackList(l2);
    printList(l);
 
    //链表头部插入新链表
    l.PushFrontList(l2);
    printList(l);
 
    //移动元素到最后
    l.MoveToBack(a1);
    printList(l);
 
    //移动元素到头部
    l.MoveToFront(a1);
    printList(l);
 
    //移动元素在某元素之后
    l.MoveAfter(b2, a1);
    printList(l);
 
    //移动元素在某元素之前
    l.MoveBefore(b2, a1);
    printList(l);
 
    //删除某元素
    l.Remove(a1);
    printList(l);
}

ring环

package main;
 
import (
    "container/ring"
    "fmt"
)
 
func printRing(r *ring.Ring) {
    r.Do(func(v interface{}) {
        fmt.Print(v.(int), " ");
    });
    fmt.Println();
}
 
func main() {
    //创建环形链表
    r := ring.New(5);
    //循环赋值
    for i := 0; i < 5; i++ {
        r.Value = i;
        //取得下一个元素
        r = r.Next();
    }
    printRing(r);
    //环的长度
    fmt.Println(r.Len());
 
    //移动环的指针
    r.Move(2);
 
    //从当前指针删除n个元素
    r.Unlink(2);
    printRing(r);
 
    //连接两个环
    r2 := ring.New(3);
    for i := 0; i < 3; i++ {
        r2.Value = i + 10;
        //取得下一个元素
        r2 = r2.Next();
    }
    printRing(r2);
 
    r.Link(r2);
    printRing(r);
}