golang 优先级队列

2021-07-16 17:34:38 阅读：205 来源： 互联网

标签：pq 优先级 heap 队列 golang item int func

container/heap

本文是 Go 标准库中 container/heap 包文档的翻译，原文地址为： https://golang.org/pkg/container/heap/

概述

包 heap 为所有实现了 heap.Interface 的类型提供堆操作。一个堆即是一棵树，这棵树的每个节点的值都比它的子节点的值要小，而整棵树最小的值位于树根（root），也即是索引 0 的位置上。

堆是实现优先队列的一种常见方法。为了构建优先队列，用户在实现堆接口时，需要让 Less() 方法返回逆序的结果，这样就可以在使用 Push 添加元素的同时，通过 Pop 移除队列中优先级最高的元素了。具体的实现请看接下来展示的优先队列例子。

示例：整数堆

// 这段代码演示了如何使用堆接口构建一个整数堆。
package main

import (
	"container/heap"
	"fmt"
)

// IntHeap 是一个由整数组成的最小堆。
type IntHeap []int

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{}) {
	// Push 和 Pop 使用 pointer receiver 作为参数，
	// 因为它们不仅会对切片的内容进行调整，还会修改切片的长度。
	*h = append(*h, x.(int))
}

func (h *IntHeap) Pop() interface{} {
	old := *h
	n := len(old)
	x := old[n-1]
	*h = old[0 : n-1]
	return x
}

// 这个示例会将一些整数插入到堆里面， 接着检查堆中的最小值，
// 之后按顺序从堆里面移除各个整数。
func main() {
	h := &IntHeap{2, 1, 5}
	heap.Init(h)
	heap.Push(h, 3)
	fmt.Printf("minimum: %d\n", (*h)[0])
	for h.Len() > 0 {
		fmt.Printf("%d ", heap.Pop(h))
	}
}

执行结果：

minimum: 1
1 2 3 5

示例：优先队列

// 这段代码演示了如何使用堆接口构建一个优先队列。
package main

import (
	"container/heap"
	"fmt"
)

// Item 是优先队列中包含的元素。
type Item struct {
	value    string // 元素的值，可以是任意字符串。
	priority int    // 元素在队列中的优先级。
	// 元素的索引可以用于更新操作，它由 heap.Interface 定义的方法维护。
	index int // 元素在堆中的索引。
}

// 一个实现了 heap.Interface 接口的优先队列，队列中包含任意多个 Item 结构。
type PriorityQueue []*Item

func (pq PriorityQueue) Len() int { return len(pq) }

func (pq PriorityQueue) Less(i, j int) bool {
	// 我们希望 Pop 返回的是最大值而不是最小值，
	// 因此这里使用大于号进行对比。
	return pq[i].priority > pq[j].priority
}

func (pq PriorityQueue) Swap(i, j int) {
	pq[i], pq[j] = pq[j], pq[i]
	pq[i].index = i
	pq[j].index = j
}

func (pq *PriorityQueue) Push(x interface{}) {
	n := len(*pq)
	item := x.(*Item)
	item.index = n
	*pq = append(*pq, item)
}

func (pq *PriorityQueue) Pop() interface{} {
	old := *pq
	n := len(old)
	item := old[n-1]
	item.index = -1 // 为了安全性考虑而做的设置
	*pq = old[0 : n-1]
	return item
}

// 更新函数会修改队列中指定元素的优先级以及值。
func (pq *PriorityQueue) update(item *Item, value string, priority int) {
	item.value = value
	item.priority = priority
	heap.Fix(pq, item.index)
}

// 这个示例首先会创建一个优先队列，并在队列中包含一些元素
// 接着将一个新元素添加到队列里面，并对其进行操作
// 最后按优先级有序地移除队列中的各个元素。
func main() {
	// 一些元素以及它们的优先级。
	items := map[string]int{
		"banana": 3, "apple": 2, "pear": 4,
	}

	// 创建一个优先队列，并将上述元素放入到队列里面，
	// 然后对队列进行初始化以满足优先队列（堆）的不变性。
	pq := make(PriorityQueue, len(items))
	i := 0
	for value, priority := range items {
		pq[i] = &Item{
			value:    value,
			priority: priority,
			index:    i,
		}
		i++
	}
	heap.Init(&pq)

	// 插入一个新元素，然后修改它的优先级。
	item := &Item{
		value:    "orange",
		priority: 1,
	}
	heap.Push(&pq, item)
	pq.update(item, item.value, 5)

    // 以降序形式取出并打印队列中的所有元素。
	for pq.Len() > 0 {
		item := heap.Pop(&pq).(*Item)
		fmt.Printf("%.2d:%s ", item.priority, item.value)
	}
}

执行结果：

05:orange 04:pear 03:banana 02:apple

Fix 函数

func Fix(h Interface, i int)

在索引 i 上的元素的值发生变化之后，重新修复堆的有序性。先修改索引 i 上的元素的值然后再执行 Fix ，跟先调用 Remove(h, i) 然后再使用 Push 操作将新值重新添加到堆里面的做法具有同等的效果，但前者所需的计算量稍微要少一些。

Fix 函数的复杂度为 O(log(n)) ，其中 n 等于 h.Len() 。

Init 函数

func Init(h Interface)

在执行任何堆操作之前，必须对堆进行初始化。 Init 操作对于堆不变性（invariants）具有幂等性，无论堆不变性是否有效，它都可以被调用。

Init 函数的复杂度为 O(n) ，其中 n 等于 h.Len() 。

Pop 函数¶

func Pop(h Interface) interface{}

Pop 函数根据 Less 的结果，从堆中移除并返回具有最小值的元素，等同于执行 Remove(h, 0) 。

Pop 函数的复杂度为 O(log(n)) ，其中 n 等于 h.Len() 。

Push 函数

func Push(h Interface, x interface{})

Push 函数将值为 x 的元素推入到堆里面，该函数的复杂度为 O(log(n)) ，其中 n 等于 h.Len() 。

Remove 函数

func Remove(h Interface, i int) interface{}

Remove 函数将移除堆中索引为 i 的元素，该函数的复杂度为 O(log(n)) ，其中 n 等于 h.Len() 。

Interface 类型

任何实现了 heap.Interface 接口的类型，都可以用作带有以下不变性的最小堆，（换句话说，这个堆在为空、已排序或者调用 Init 之后，应该具有以下性质）：

!h.Less(j, i) for 0 <= i < h.Len() and 2*i+1 <= j <= 2*i+2 and j < h.Len()

注意，这个接口中的 Push 和 Pop 都是由 heap 包的实现负责调用的。因此用户在向堆添加元素又或者从堆中移除元素时，需要使用 heap.Push 以及 heap.Pop ：

type Interface interface {
    sort.Interface
    Push(x interface{}) // 将 x 添加为元素 Len()
    Pop() interface{}   // 移除并返回元素 Len() - 1
}

标签：pq,优先级,heap,队列,golang,item,int,func
来源： https://www.cnblogs.com/dream397/p/15021120.html

本站声明： 1. iCode9 技术分享网（下文简称本站）提供的所有内容，仅供技术学习、探讨和分享；
2. 关于本站的所有留言、评论、转载及引用，纯属内容发起人的个人观点，与本站观点和立场无关；
3. 关于本站的所有言论和文字，纯属内容发起人的个人观点，与本站观点和立场无关；
4. 本站文章均是网友提供，不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属；如您发现该文章侵犯了您的权益，可联系我们第一时间进行删除；
5. 本站为非盈利性的个人网站，所有内容不会用来进行牟利，也不会利用任何形式的广告来间接获益，纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。

ICode9