优先队列(PriorityQueue)

一、概念

优先队列 = 自带自动排序的队列 ,出队顺序不再是先进先出 (FIFO),而是优先级最高元素先出队 。底层常用:堆 (Heap,二叉堆),入队 / 出队时间复杂度 O(logn)。

两种堆

  1. 小顶堆(默认 Java/Python):堆顶是最小值,最小先弹出
  2. 大顶堆:堆顶是最大值,最大先弹出

二、Java PriorityQueue

java 复制代码
// 小顶堆,队首最小
PriorityQueue<Integer> minHeap = new PriorityQueue<>();
minHeap.add(5);
minHeap.add(2);
minHeap.add(7);
// poll()弹出堆顶最小值:2
System.out.println(minHeap.poll());

// 大顶堆(自定义比较器)
PriorityQueue<Integer> maxHeap = new PriorityQueue<>((a,b)->b-a);
方法 作用
add(E e) 入队,满抛异常
offer(E e) 入队,满返回 false
peek() 查看堆顶,不删除,空返回 null
poll() 弹出堆顶,空返回 null
size() 元素个数

三、Python heapq(默认小顶堆)

Python没有原生大顶堆,存负数模拟大顶堆

python 复制代码
import heapq
arr = [3,1,4]
heapq.heapify(arr) # 原地建堆 O(n)
heapq.heappop(arr) # 弹出最小1
heapq.heappush(arr,2)

# 大顶堆:存入负数
maxheap = []
heapq.heappush(maxheap,-5)
print(-heapq.heappop(maxheap))

四、经典使用场景

  1. TopK 问题:求数组前 K 大 → 用容量 K 小顶堆
  2. 多路归并、Dijkstra 最短路径(每次取最短边)
  3. 任务调度、贪心算法

五、时间复杂度

  • 建堆:O(n)
  • 插入 / 删除堆顶:O(logn)
相关推荐
hai3152475431 小时前
有规则的AI编制操作系统演进过程展示
人工智能·程序人生·算法·逻辑回归·创业创新
数据仓库搬砖人1 小时前
SHAP 详解:从博弈论原理到 XGBoost 实战
算法
老鱼说AI1 小时前
统计学习方法第七章:支持向量机精讲(超硬核长文深入预警!)
人工智能·深度学习·神经网络·算法·机器学习·支持向量机·学习方法
容器魔方1 小时前
KubeEdge SIG AI: 基于KubeEdge-Ianvs的大模型联邦微调算法
大数据·人工智能·算法·云原生·容器·云计算
列星随旋1 小时前
矩阵快速幂
java·算法·矩阵
z200509301 小时前
今日算法(回溯全排列)
c++·算法·leetcode
Boom_Shu1 小时前
构造函数程序
数据结构·算法
MicroTech20251 小时前
微算法科技(NASDAQ: MLGO)量子安全与区块链:量子神经网络QNN赋能动态共识与量子密钥分发
科技·算法·安全
Lucky_ldy1 小时前
数据结构从入门到精通:链表
数据结构·链表