0

0

如何利用Java的PriorityQueue实现中位数计算_双堆模型实战

P粉602998670

P粉602998670

发布时间:2026-03-16 13:52:31

|

641人浏览过

|

来源于php中文网

原创

不能直接用PriorityQueue求中位数,因其仅支持O(1)获取堆顶,无法高效取得第⌊n/2⌋小元素;须用大顶堆存较小一半、小顶堆存较大一半,并始终维持大小差≤1且堆顶有序。

如何利用java的priorityqueue实现中位数计算_双堆模型实战

为什么不能直接用 PriorityQueue 求中位数?

因为 PriorityQueue 只能高效获取堆顶(最小或最大),无法随机访问中间元素,更不能在 O(1) 或 O(log n) 时间内拿到第 ⌊n/2⌋ 小的值。中位数要求动态维护“中间位置”,必须靠两个堆协同——一个大顶堆存较小一半,一个小顶堆存较大一半。

怎么初始化大小顶堆才能保证平衡?

Java 没有原生大顶堆,得用 PriorityQueue 配合 Comparator.reverseOrder();小顶堆用默认构造。关键不是初始容量,而是始终维持:maxHeap.size() == minHeap.size()maxHeap.size() == minHeap.size() + 1

实操建议:

  • 始终让 maxHeap(大顶堆)保存较小的一半,所以它的堆顶就是当前中位数的“左候选”
  • 每次 addNum(int num) 后,先塞进 maxHeap,再把它的堆顶弹出塞进 minHeap,最后检查并调整两堆大小关系
  • 避免直接比较 num 和堆顶再决定插入哪个堆——容易漏掉边界情况(比如堆为空时)

addNum 过程中常见的堆失衡错误

典型现象:中位数突然跳变、偶数个数时返回错值、NullPointerException(空堆调 peek())。

立即学习Java免费学习笔记(深入)”;

根本原因往往是没统一处理“先入 maxHeap → 调整 → 再平衡”的流程,而是写成条件分支判断。正确做法是固定流水线:

OpenJobs AI
OpenJobs AI

AI驱动的职位搜索推荐平台

下载
  • 把新数加进 maxHeap
  • 立刻弹出 maxHeap.peek(),放进 minHeap
  • 如果 minHeap.size() > maxHeap.size(),就把 minHeap.poll() 塞回 maxHeap
  • 所有 peek() 前加非空判断(尤其首次添加时)

示例片段:

if (!maxHeap.isEmpty() && num > maxHeap.peek()) { ... }
是错的——这会破坏双堆不变量,应全程走统一调度逻辑。

取中位数时 findMedian 的性能和精度陷阱

看似简单,但两个细节极易出错:一是整数除法导致精度丢失(如 (a + b) / 2 在 a,b 都是 int 时截断),二是堆大小判断逻辑写反。

实操要点:

  • 返回类型必须是 double,计算时至少一个操作数转 double(double) maxHeap.peek() / 1.0(maxHeap.peek() + minHeap.peek()) / 2.0
  • 只在 maxHeap.size() == minHeap.size() 时取两堆顶平均;否则直接返回 maxHeap.peek()(因约定它多一个)
  • 别依赖 size() 的绝对值做分支,而要用相对关系——比如 maxHeap.size() - minHeap.size() == 1maxHeap.size() == 1 更鲁棒

双堆模型真正难的不是代码长度,而是每一步都得守住“maxHeap 顶 ≤ minHeap 顶”且“大小差 ≤ 1”这两个不变量。漏一次校验,后续全乱。

热门AI工具

更多
DeepSeek
DeepSeek

幻方量化公司旗下的开源大模型平台

豆包大模型
豆包大模型

字节跳动自主研发的一系列大型语言模型

WorkBuddy
WorkBuddy

腾讯云推出的AI原生桌面智能体工作台

腾讯元宝
腾讯元宝

腾讯混元平台推出的AI助手

文心一言
文心一言

文心一言是百度开发的AI聊天机器人,通过对话可以生成各种形式的内容。

讯飞写作
讯飞写作

基于讯飞星火大模型的AI写作工具,可以快速生成新闻稿件、品宣文案、工作总结、心得体会等各种文文稿

即梦AI
即梦AI

一站式AI创作平台,免费AI图片和视频生成。

ChatGPT
ChatGPT

最最强大的AI聊天机器人程序,ChatGPT不单是聊天机器人,还能进行撰写邮件、视频脚本、文案、翻译、代码等任务。

相关专题

更多
string转int
string转int

在编程中,我们经常会遇到需要将字符串(str)转换为整数(int)的情况。这可能是因为我们需要对字符串进行数值计算,或者需要将用户输入的字符串转换为整数进行处理。php中文网给大家带来了相关的教程以及文章,欢迎大家前来学习阅读。

1071

2023.08.02

int占多少字节
int占多少字节

int占4个字节,意味着一个int变量可以存储范围在-2,147,483,648到2,147,483,647之间的整数值,在某些情况下也可能是2个字节或8个字节,int是一种常用的数据类型,用于表示整数,需要根据具体情况选择合适的数据类型,以确保程序的正确性和性能。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

617

2024.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

335

2025.08.29

C++中int的含义
C++中int的含义

本专题整合了C++中int相关内容,阅读专题下面的文章了解更多详细内容。

235

2025.08.29

c++怎么把double转成int
c++怎么把double转成int

本专题整合了 c++ double相关教程,阅读专题下面的文章了解更多详细内容。

335

2025.08.29

C++中int、float和double的区别
C++中int、float和double的区别

本专题整合了c++中int和double的区别,阅读专题下面的文章了解更多详细内容。

108

2025.10.23

堆和栈的区别
堆和栈的区别

堆和栈的区别:1、内存分配方式不同;2、大小不同;3、数据访问方式不同;4、数据的生命周期。本专题为大家提供堆和栈的区别的相关的文章、下载、课程内容,供大家免费下载体验。

448

2023.07.18

堆和栈区别
堆和栈区别

堆(Heap)和栈(Stack)是计算机中两种常见的内存分配机制。它们在内存管理的方式、分配方式以及使用场景上有很大的区别。本文将详细介绍堆和栈的特点、区别以及各自的使用场景。php中文网给大家带来了相关的教程以及文章欢迎大家前来学习阅读。

606

2023.08.10

chatgpt使用指南
chatgpt使用指南

本专题整合了chatgpt使用教程、新手使用说明等等相关内容,阅读专题下面的文章了解更多详细内容。

0

2026.03.16

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
相关推荐
/
热门推荐
/
最新课程
Kotlin 教程
Kotlin 教程

共23课时 | 4.5万人学习

C# 教程
C# 教程

共94课时 | 11.5万人学习

Java 教程
Java 教程

共578课时 | 83.2万人学习

关于我们 免责申明 举报中心 意见反馈 讲师合作 广告合作 最新更新
php中文网:公益在线php培训,帮助PHP学习者快速成长!
关注服务号 技术交流群
PHP中文网订阅号
每天精选资源文章推送

Copyright 2014-2026 https://www.php.cn/ All Rights Reserved | php.cn | 湘ICP备2023035733号