算法通关村第十四关—数据流的中位数(黄金)

数据流中中位数的问题

LeetCode295,中位数是有序列表中间的数。如果列表长度是偶数,中位数则是中间两个数的平均值。

例如:[2,3,4]的中位数是3

2,3\]的中位数是(2+3)/2=2.5 实现 MedianFinder 类: * MedianFinder() 初始化 MedianFinder 对象。 * void addNum(int num) 将数据流中的整数 num 添加到数据结构中。 * double findMedian() 返回到目前为止所有元素的中位数。与实际答案相差 10-5 以内的答案将被接受。 ![image.png](https://file.jishuzhan.net/article/1748246899825053698/b7bdb0c9db8eadfd7f4f97632e23e1a6.webp)  分析这是一道比较难的题目了,如果没专门学过,很难在面试时想到。  中位数的题,我们一般都可以用大顶堆+小顶堆来求解,下面我们通过直观的例子解释一下怎么做。 小顶堆(minHeap):存储所有元素中较大的一半,堆顶存储的是其中最小的数。 大顶堆(maxHeap):存储所有元素中较小的一半,堆顶存储的是其中最大的数。  相当于,把所有元素分成了大和小两半,而我们计算中位数,只需要大的那半的最小值和小的那半的最大值即可。比如,我们依次添加\[1,2,3,4,5\],砍成两半之后为\[1,2\]和\[3,4,5\],我们只要能快速的找到2和3即可。  下面看看使用两个堆它们是怎么变化的: 1.添加1,进入到minHeap中,中位数为1:![截屏2024-01-15 07.27.57.png](https://file.jishuzhan.net/article/1748246899825053698/d9317ef20bd61724017e5f71cd8c8b1f.webp) 2.添加2,它比minHeap堆顶元素1大,进入minHeap,同时,minHeap中元素超过了所有元素总和的一半,所以,要平衡一下,分一个给maxHeap,中位数为(1+2)/2.0=1.5:![截屏2024-01-15 07.28.05.png](https://file.jishuzhan.net/article/1748246899825053698/9a05ea8b3f7dd37e8efe4a4c79cea033.webp) 添加3,它比minHeap堆顶元素2大,进入minHeap,中位数为2:![截屏2024-01-15 07.28.13.png](https://file.jishuzhan.net/article/1748246899825053698/e049eab7e918a232a4c1b4b3200920ef.webp) 添加4,它比minHeap堆顶元素2大,进入minHeap,同时,minHeap中元素超过了所有元素总和的一半,所以,要平衡一下,分一个给maxHeap,中位数为(2+3)/2.0=2.5:![截屏2024-01-15 07.28.33.png](https://file.jishuzhan.net/article/1748246899825053698/6cf87860a33db4a588b929a896440e84.webp) 5.添加5,它比minHeap堆J顶元素3大,进入minHeap,中位数为3:![截屏2024-01-15 07.28.40.png](https://file.jishuzhan.net/article/1748246899825053698/9630c17c7f8188895dab7bc3f6bc94a0.webp)  Java中的堆(即优先级队列)是使用完全二叉树实现的,我们这里的图也是以完全二叉树为例。理解了上述的过程,看代码就比较简单了  代码如下 ```java class MedianFinder { PriorityQueue queleft; PriorityQueue queright; public MedianFinder() { queleft = new PriorityQueue((a, b) -> (b - a));//中位数左边是大顶堆 queright = new PriorityQueue();//中位数右边是小顶堆 } public void addNum(int num) { //添加元素 if(queleft.isEmpty() || num <= queleft.peek()){ queleft.offer(num); if(queleft.size() > queright.size() + 1){ //queleft最多比queright多一个元素 queright.offer(queleft.poll()); } } else{ queright.offer(num); if(queright.size() > queleft.size()){ queleft.offer(queright.poll()); } } } public double findMedian() { if(queleft.size() > queright.size()){//奇数情况 return 1.0 * queleft.peek(); } else return (queleft.peek() + queright.peek()) / 2.0; } } ```

相关推荐
再睡一夏就好1 分钟前
从硬件角度理解“Linux下一切皆文件“,详解用户级缓冲区
linux·服务器·c语言·开发语言·学习笔记
愛~杦辷个訾1 小时前
芋道项目,商城模块数据表结构
java·sql·芋道·yudao-cloud·芋道商城
TIF星空1 小时前
【使用 C# 获取 USB 设备信息及进行通信】
开发语言·经验分享·笔记·学习·microsoft·c#
Smile丶凉轩3 小时前
Qt 界面优化(绘图)
开发语言·数据库·c++·qt
reasonsummer4 小时前
【办公类-100-01】20250515手机导出教学照片,自动上传csdn+最大化、最小化Vs界面
开发语言·python
S01d13r4 小时前
LeetCode 解题思路 48(编辑距离、只出现一次的数字)
算法·leetcode·职场和发展
C_Liu_4 小时前
C语言:深入理解指针(5)
java·c语言·算法
small_wh1te_coder4 小时前
从经典力扣题发掘DFS与记忆化搜索的本质 -从矩阵最长递增路径入手 一步步探究dfs思维优化与编程深度思考
c语言·数据结构·c++·stm32·算法·leetcode·深度优先
枫景Maple4 小时前
LeetCode 45. 跳跃游戏 II(中等)
算法·leetcode
এ᭄画画的北北5 小时前
力扣-236.二叉树的最近公共祖先
算法·leetcode