golang实现优先队列,以前写的一个简单例子,现上传备份
package test
import (
"container/heap"
"fmt"
)
func Test() {
// 测试一下任务队列
// 1、首先测试是标准任务队列,队列之中的元素是结构体Person
pq := &AgePQ{}
heap.Init(pq)
ages := []int{2, 3, 1, 6, 5}
for i := range ages {
tem := &Person{Age: ages[i]}
heap.Push(pq, tem)
}
// 查看pq情况
fmt.Println(heap.Pop(pq).(*Person).Age) //6
fmt.Println(heap.Pop(pq).(*Person).Age) //5
fmt.Println(heap.Pop(pq).(*Person).Age) //3
fmt.Println(heap.Pop(pq).(*Person).Age) //2
fmt.Println(heap.Pop(pq).(*Person).Age) //1
// 这里要注意一下pop方法是在队列的顶进行的操作,如果是升序,则pop出的是最小的值,降序则pop出的是最大的值
fmt.Println("—————我是传说中的分割线————")
// 2、接下来测试的是int元素所组成的优先队列
pq1 := &PQ_of_int{}
heap.Init(pq1)
for i := range ages {
heap.Push(pq1, ages[i])
}
fmt.Println(heap.Pop(pq1).(int)) //1
fmt.Println(heap.Pop(pq1).(int)) //2
fmt.Println(heap.Pop(pq1).(int)) //3
fmt.Println(heap.Pop