【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这样的顺序,可能这里有些说法我不是很清楚

相关推荐
yangzhi_emo18 分钟前
ES6笔记2
开发语言·前端·javascript
ai小鬼头43 分钟前
Ollama+OpenWeb最新版0.42+0.3.35一键安装教程,轻松搞定AI模型部署
后端·架构·github
emplace_back1 小时前
C# 集合表达式和展开运算符 (..) 详解
开发语言·windows·c#
jz_ddk1 小时前
[学习] C语言数学库函数背后的故事:`double erf(double x)`
c语言·开发语言·学习
萧曵 丶1 小时前
Rust 所有权系统:深入浅出指南
开发语言·后端·rust
xiaolang_8616_wjl2 小时前
c++文字游戏_闯关打怪2.0(开源)
开发语言·c++·开源
收破烂的小熊猫~2 小时前
《Java修仙传:从凡胎到码帝》第四章:设计模式破万法
java·开发语言·设计模式
老任与码2 小时前
Spring AI Alibaba(1)——基本使用
java·人工智能·后端·springaialibaba
nananaij2 小时前
【Python进阶篇 面向对象程序设计(3) 继承】
开发语言·python·神经网络·pycharm