1. 优先队列简介
优先队列(Priority Queue):一种特殊的队列。在优先队列中,元素被赋予优先级,当访问队列元素时,具有最高优先级的元素最先删除
与普通队列最大的不同:出队顺序,在优先队列中,具有最高优先级的元素最先出队
普通队列的出队顺序跟入队顺序相关,符合**先进先出(First in, First out)**的规则。
而优先队列的出队顺序跟入队顺序无关,优先队列是按照元素的优先级来决定出队顺序的。优先级高的元素优先出队,优先级低的元素后出队。优先队列符合 最高级先出(First in, Largest out) 的规则
2. 适用场景
- 数据压缩:赫夫曼编码算法
- 最短路径算法:Dijkstra 算法
- 最小生成树算法:Prim 算法
- 任务调度器:根据优先级执行系统任务
- 事件驱动仿真:顾客排队算法
- 选择问题:查找第 k 个最小元素
3. 优先队列的实现
由上面的定义可知,优先队列与一般队列的不同在于入队和出队操作
而优先队列的实现方式也有很多种,除了使用**数组(顺序存储)实现与链表(链式存储)**实现之外,我们最常用的是使用 二叉堆结构实现优先队列。
- 数组(顺序存储)实现优先队列:入队操作直接插入到数组队尾,时间复杂度为 O(1)。出队操作需要遍历整个数组,找到优先级最高的元素,返回并删除该元素,时间复杂度为O(n)
- 链表(链式存储)实现优先队列:链表中的元素按照优先级排序,入队操作需要为待插入元素创建节点,并在链表中找到合适的插入位置,时间复杂度为O(n)。出队操作直接返回链表队头元素,并删除队头元素,时间复杂度为O(1)。
- 二叉堆结构实现优先队列:构建一个二叉堆结构,二叉堆按照优先级进行排序。入队操作就是将元素插入到二叉堆中合适位置,时间复杂度为O(logn)。吹对操作则返回二叉堆中优先级最大节点并删除,时间复杂度也是O(logn)
综上所述,优先队列使用二叉堆实现效率较高
4. 二叉堆实现的优先队列
-
定义:
- 二叉堆:符合以下两个条件之一的完全二叉树
1. 大顶堆:根节点值 ≥ 子节点值
2. 小顶堆:根节点值 ≤ 子节点值
- 二叉堆:符合以下两个条件之一的完全二叉树
-
二叉堆的基本操作:
二叉堆主要涉及到两个操作:堆调整方法和将数组构建为二叉堆方法- 堆调整方法 heapAdjust:把移走了最大值元素以后的剩余元素组成的序列再构造为一个新的堆积
具体步骤如下:
1. 从根节点开始,自上而下地调整节点的位置,使其成为堆积。即把序号为 i 的节点与其左子树节点(序号为 2 * i)、右子树节点(序号为 2 * i + 1)中值最大的节点交换位置
2. 因为交换了位置,使得当前节点的左右子树原有的堆积特性被破坏。于是,从当前节点的左右子树节点开始,自上而下继续进行类似的调整
3. 如此下去直到整棵完全二叉树成为一个大顶堆 - 将数组构建为二叉堆方法(初始堆建立方法) heapify
步骤如下:
1. 如果原始序列对应的完全二叉树(不一定是堆)的深度为 d,则从 d - 1 层最右侧分支节点(序号为 ⌊n/2⌋)开始,初始时令 i = ⌊n/2⌋,调用堆调整算法
2. 每调用一次堆调整算法,执行一次 i = i - 1,直到 i == 1 时,再调用一次,就把原始数组构建为了一个二叉堆
- 堆调整方法 heapAdjust:把移走了最大值元素以后的剩余元素组成的序列再构造为一个新的堆积
-
优先队列的基本操作:
- 入队heappush:
- 先将待插入元素 value 插入到数组 nums 末尾
- 如果完全二叉树的深度为 d,则从 d - 1 层开始最右侧分支节点(序号为 ⌊n/2⌋)开始,初始时令 i = ⌊n/2⌋,从下向上依次查找插入位置
- 遇到 value 小于当前根节点时,将其插入到当前位置。否则继续向上寻找插入位置
- 如果找到插入位置或者到达根位置,将 value 插入该位置
- 出队heappop:
- 交换数组 nums 首尾元素,此时 nums 尾部就是值最大(优先级最高)的元素,将其从 nums 中弹出,并保存起来
- 弹出后,对 nums 剩余元素调用堆调整算法,将其调整为大顶堆
- 入队heappush:
-
手写二叉堆实现优先队列
- heapAdjust:将完全二叉树调整为二叉堆
- heapify: 将数组构建为二叉堆方法(初始堆建立方法)
- heappush:向堆中添加元素,也是优先队列的入队操作
- heappop:删除堆顶元素,也是优先队列的出队操作,弹出优先队列中优先级最高的元素
- heapSort:堆排序
class Heapq: # 堆调整方法:调整为大顶堆 def heapAdjust(self, nums: [int], index: int, end: int): left = index * 2 + 1 right = left + 1 while left <= end: # 当前节点为非叶子结点 max_index = index if nums[left] > nums[max_index]: max_index = left if right <= end and nums[right] > nums