标签:right Offer offer 中位数 PriorityQueue 63 size left
1. 本题知识点
堆
2. 题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用 Insert() 方法读取数据流,使用 GetMedian() 方法获取当前读取数据的中位数。
3. 解题思路
- 创建一个大顶堆用于存储输入数字中较小一半,即中位数左边的一半
- 创建一个小顶堆用于存储输入数字中较大一半,即中位数右边的一半
- 保证最大堆中最大的元素 < 最小堆中最小元素,并且他们之前的元素个数相差不大于 1
- 那么中位数就可以通过这两个堆的堆顶元素获得
4.代码
public class Solution {
// 总个数
private int size = 0;
// 小顶堆
private PriorityQueue<Integer> left = new PriorityQueue<>((o1, o2) -> o2 -o1);
// 大顶堆
private PriorityQueue<Integer> right = new PriorityQueue<>();
public void Insert(Integer num) {
if (size % 2 == 0) {
left.offer(num);
right.offer(left.poll());
} else {
right.offer(num);
left.offer(right.poll());
}
size++;
}
public Double GetMedian() {
if (size % 2 == 0) {
return (left.peek() + right.peek()) / 2.0;
} else {
return (double) right.peek();
}
}
}
标签:right,Offer,offer,中位数,PriorityQueue,63,size,left 来源: https://blog.csdn.net/bm1998/article/details/113850744
本站声明: 1. iCode9 技术分享网(下文简称本站)提供的所有内容,仅供技术学习、探讨和分享; 2. 关于本站的所有留言、评论、转载及引用,纯属内容发起人的个人观点,与本站观点和立场无关; 3. 关于本站的所有言论和文字,纯属内容发起人的个人观点,与本站观点和立场无关; 4. 本站文章均是网友提供,不完全保证技术分享内容的完整性、准确性、时效性、风险性和版权归属;如您发现该文章侵犯了您的权益,可联系我们第一时间进行删除; 5. 本站为非盈利性的个人网站,所有内容不会用来进行牟利,也不会利用任何形式的广告来间接获益,纯粹是为了广大技术爱好者提供技术内容和技术思想的分享性交流网站。