【golang】使用container/heap官方包实现一个优先队列

golang实现优先队列,以前写的一个简单例子,现上传备份

Go 复制代码
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(pq1).(int)) //5
	fmt.Println(heap.Pop(pq1).(int)) //6
	// 测试结果显示,这两个类型的优先队列都是能完成的
}

// 结构体优先队列
type Person struct {
	Age int
}
type AgePQ []*Person

func (pq AgePQ) Len() int {
	return len(pq)
}
func (pq AgePQ) Less(i, j int) bool {
	// 我们这按照一个降序
	return pq[i].Age > pq[j].Age
}
func (pq AgePQ) Swap(i, j int) {
	pq[i], pq[j] = pq[j], pq[i]
}

// 两个队列独有的方法
func (pq *AgePQ) Push(x any) {
	item := x.(*Person)
	*pq = append(*pq, item)
}
func (pq *AgePQ) Pop() any {
	if pq.Len() == 0 {
		return nil
	}
	last := (*pq)[pq.Len()-1]
	*pq = (*pq)[:(pq.Len() - 1)]
	return last
}

// 简单的int组成的优先队列
type PQ_of_int []int

func (pq PQ_of_int) Len() int {
	return len(pq)
}
func (pq PQ_of_int) Less(i, j int) bool {
	// 这里是一个升序
	return pq[i] < pq[j]
}
func (pq PQ_of_int) Swap(i, j int) {
	pq[i], pq[j] = pq[j], pq[i]
}
func (pq *PQ_of_int) Push(x any) {
	*pq = append(*pq, x.(int))
}
func (pq *PQ_of_int) Pop() any {
	if pq.Len() == 0 {
		return nil
	}
	last := (*pq)[pq.Len()-1]
	*pq = (*pq)[:pq.Len()-1]
	return last
}

主要测试了两个优先队列,一个队列是针对Person结构体,第二个队列单纯就是int切片

需要注意以下几个点:

1、pop和push行为最好不要直接调用,而是借助heap.Pop()这样的方法

2、pop行为可能在js当中表示删除数组的最后一个值,这里pop行为是可以自定义的,我将pop行为定义成删除队列最后的值,而真实运行起来,却是删除队列的Top值,比如这里的PQ_of_int,我在Less方法之中定义的是一个升序,但是最后pop出来的确实12356这样的顺序,可能这里有些说法我不是很清楚

相关推荐
我命由我123457 分钟前
PDFBox - PDDocument 与 byte 数组、PDF 加密
java·服务器·前端·后端·学习·java-ee·pdf
花哥码天下11 分钟前
Oracle下载JDK无需登录
java·开发语言
考虑考虑18 分钟前
go格式化时间
后端·go
早点.早点.26 分钟前
QT登陆界面
开发语言·qt
楼田莉子29 分钟前
C++学习:异常及其处理
开发语言·c++·学习·visual studio
fsnine31 分钟前
Python Web框架对比与模型部署
开发语言·前端·python
星球奋斗者42 分钟前
计算机方向如何才能更好的找到工作?(成长心得)
java·后端·考研·软件工程·改行学it
海梨花1 小时前
【八股笔记】SSM
java·开发语言·笔记·后端·面试·框架
JAVA学习通1 小时前
OJ竞赛平台----C端题目列表
java·开发语言·jvm·vue.js·elasticsearch
IT_陈寒1 小时前
Redis性能翻倍的7个冷门技巧:从P5到P8都在偷偷用的优化策略!
前端·人工智能·后端