什么是优先级队列 如何用代码手动实现一个优先级队列

优先级队列 是一种用于维护一组元素构成的集合的数据结构，其中每个元素都有关联的“优先级”。在优先级队列中，元素被按优先级进行排序，规则是预先设定的、可以按照元素值的大小排列、也可以按照元素的时间戳等其他标准排列。添加或删除元素的操作遵循优先级确定的顺序：最高优先级的元素最先被移除。这使得优先级队列成为一种非常适用于任务调度、事件驱动的存储、汇总和处理数据的高效工具。

为了详细地了解如何实现一个优先级队列，我们将对它的一种实现方式——二叉堆（Binary Heap） 进行探讨。二叉堆是一种用数组实现的二叉树结构，它可分为最大堆和最小堆，在最大堆中，任意节点的值不小于其子节点的值；在最小堆中，任意节点的值不大于其子节点的值。二叉堆的这一性质使得堆根节点存放的是整个堆中的最大值或最小值，这一特点正符合优先级队列的要求。

一、实现优先级队列的步骤

在了解二叉堆之后，我们首先需要定义优先级队列的操作，主要包括插入元素、删除元素、以及查看队列中的最高优先级元素。

二、优先级队列的数据结构选择

二叉堆是实现优先级队列的常见选择。我们将选择最小堆来表示一个优先级队列。在这里，较小值表示较高优先级。

三、优先级队列的核心功能

优先级队列通常具有以下基本操作：

插入操作（PUSH）

向优先级队列添加新元素时，我们将元素添加到二叉堆的末尾，然后将其向上移动到其正确的位置以维护堆的属性。

删除操作（POP）

删除操作涉及移除具有最高优先级的元素，通常是位于堆顶的元素。删除后，会将堆中最后一个元素移至顶部，然后向下移动以维护堆的属性。

查看操作（PEEK）

这是一个非常简单的操作，在最小堆的优先级队列中，我们只需要访问堆顶的元素即可，这个元素具有最高的优先级。

四、具体实现优先级队列的代码

现在，我们准备用代码实现一个优先级队列。在这里，我们使用Python语言进行阐述。

初始化堆结构（INITIALIZATION）

在Python中，我们可以使用列表来模拟二叉堆结构：

class PriorityQueue:

    def __init__(self):
        self.heap = []
    # 这里省略其他方法的实现

插入元素（PUSH）

元素首先被添加到列表的末尾，然后通过比较与父节点的大小关系进行上浮操作，以维护最小堆性质。

def push(self, item):

    # 添加元素到堆末尾
    self.heap.append(item)
    self._float_up(len(self.heap) - 1)
def _float_up(self, index):
    # 通过交换元素使新增节点上浮到正确位置
    parent = (index - 1) // 2
    if index > 0 and self.heap[index] < self.heap[parent]:
        self.heap[index], self.heap[parent] = self.heap[parent], self.heap[index]
        self._float_up(parent)

移除元素（POP）

移除最优先的元素（位于堆顶的元素），将最后一个元素移至堆顶，然后进行下沉操作维护最小堆性质。

def pop(self):

    if len(self.heap) > 1:
        # 将堆顶元素与最后一个元素互换
        self.heap[0], self.heap[-1] = self.heap[-1], self.heap[0]
        # 移除互换后处于末尾的最大元素
        item = self.heap.pop()
        # 从新的堆顶下沉调整堆
        self._sink_down(0)
    elif len(self.heap) == 1:
        # 如果只有一个元素，直接移除
        item = self.heap.pop()
    else:
        rAIse IndexError('pop from an empty priority queue')
    return item
def _sink_down(self, index):
    left = 2 * index + 1
    right = 2 * index + 2
    smallest = index
    # 比较左右子节点找出最小值
    if left < len(self.heap) and self.heap[left] < self.heap[smallest]:
        smallest = left
    if right < len(self.heap) and self.heap[right] < self.heap[smallest]:
        smallest = right
    # 如果当前节点不是最小值，进行交换并继续下沉
    if smallest != index:
        self.heap[index], self.heap[smallest] = self.heap[smallest], self.heap[index]
        self._sink_down(smallest)

查看队列中的最高优先级元素（PEEK）

def peek(self):

    if self.heap:
        return self.heap[0]
    else:
        return None

五、优先级队列的高级功能

除了以上基本操作之外，优先级队列还可以实现更高级的功能，如批量创建、合并队列、调整元素优先级等。

批量创建堆（HEAPIFY）

def heapify(lst):

    n = len(lst)
    # 最后一个有子节点的节点开始向下调整
    for i in range(n // 2 - 1, -1, -1):
        _heapify_helper(lst, n, i)
def _heapify_helper(lst, heap_size, i):
    left = 2 * i + 1
    right = 2 * i + 2
    smallest = i
    if left < heap_size and lst[left] < lst[smallest]:
        smallest = left
    if right < heap_size and lst[right] < lst[smallest]:
        smallest = right
    if smallest != i:
        lst[i], lst[smallest] = lst[smallest], lst[i]
        _heapify_helper(lst, heap_size, smallest)

队列合并（MERGE）

def merge(self, other_queue):

    # 将两个优先级队列合并
    self.heap.extend(other_queue.heap)
    self.heapify(self.heap)

以上就是手动实现一个优先级队列的基本步骤和示例代码。实际情况中，优先级队列的实现可能还会更复杂，因为还需要考虑到线程安全性、不同类型元素的比较方法、自定义优先级比较逻辑等因素。然而，以上给出的示例为理解和实现一个基本的优先级队列提供了一个良好的出发点。

相关问答FAQs：

1. 什么是优先级队列？
优先级队列是一种特殊的队列数据结构，其中每个元素都具有与之相关的优先级。优先级高的元素会先被处理，而优先级相同的元素则按照它们进入队列的顺序进行处理。优先级队列常用于需要根据某种规则或条件对元素进行排序和处理的场景。

2. 如何用代码手动实现一个优先级队列？
在代码中手动实现一个优先级队列需要利用适当的数据结构和算法来保证元素的优先级和顺序的维护。以下是一个示例实现，基于最小堆（Min-Heap）的优先级队列：

• 创建一个空的数组作为最小堆（Heap）来存储元素。
• 定义一个函数来插入元素到最小堆中，同时维护元素的优先级。
• 定义一个函数来删除并返回优先级最高的元素，同时重新调整最小堆。
• 创建一个函数来获取当前优先级最高的元素，但不删除它。

实现一个优先级队列的代码示例：

class PriorityQueue:
  def __init__(self):
    self.heap = []
    
  def insert(self, item):
    self.heap.append(item)
    self._percolate_up(len(self.heap) - 1)
    
  def delete_min(self):
    if len(self.heap) == 0:
      return None
    if len(self.heap) == 1:
      return self.heap.pop()
      
    min_item = self.heap[0]
    self.heap[0] = self.heap.pop()
    self._percolate_down(0)
    return min_item
    
  def get_min(self):
    if len(self.heap) == 0:
      return None
    return self.heap[0]
    
  # Helper methods for percolating up and down
  def _percolate_up(self, current_index):
    parent_index = (current_index - 1) // 2
    while parent_index >= 0 and self.heap[current_index] < self.heap[parent_index]:
      self.heap[current_index], self.heap[parent_index] = self.heap[parent_index], self.heap[current_index]
      current_index = parent_index
      parent_index = (current_index - 1) // 2
    
  def _percolate_down(self, current_index):
    child_index = 2 * current_index + 1
    while child_index < len(self.heap):
      if child_index + 1 < len(self.heap) and self.heap[child_index + 1] < self.heap[child_index]:
        child_index += 1
      
      if self.heap[current_index] <= self.heap[child_index]:
        break
        
      self.heap[current_index], self.heap[child_index] = self.heap[child_index], self.heap[current_index]
      current_index = child_index
      child_index = 2 * current_index + 1

请注意，这只是一个示例实现，实际应用中可能会有更多的细节和优化。你可以根据自己的需求进行修改和扩展。

3. 哪些场景适合使用优先级队列？
优先级队列适用于需要按照优先级处理元素的各种场景，包括但不限于：

• 任务调度：根据任务的优先级调度执行顺序。
• 路由算法：选择下一跳的路由器或机器时考虑优先级。
• 事件管理：按照事件的紧急程度处理事件队列。
• 搜索算法：利用优先级队列进行广度优先搜索（BFS）或最短路径搜索等。

通过合理地使用优先级队列，可以提高算法的效率和性能，并简化代码实现。

最后建议，企业在引入信息化系统初期，切记要合理有效地运用好工具，这样一来不仅可以让公司业务高效地运行，还能最大程度保证团队目标的达成。同时还能大幅缩短系统开发和部署的时间成本。特别是有特定需求功能需要定制化的企业，可以采用我们公司自研的企业级低代码平台：织信Informat。 织信平台基于数据模型优先的设计理念，提供大量标准化的组件，内置AI助手、组件设计器、自动化（图形化编程）、脚本、工作流引擎（BPMN2.0）、自定义API、表单设计器、权限、仪表盘等功能，能帮助企业构建高度复杂核心的数字化系统。如ERP、MES、CRM、PLM、SCM、WMS、项目管理、流程管理等多个应用场景，全面助力企业落地国产化/信息化/数字化转型战略目标。