题目描述
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
解析:本题考察大顶堆和小顶堆:
用小顶堆存放后半部分的数(大),大顶堆存放前半部分的数(小),这样两个堆的堆顶就是中间的数字。问题是该如何存放。
数先放到大顶堆,再从大顶堆中弹出,送入小顶堆,这样可以保证大顶堆中的数永远比小堆小。
import java.util.PriorityQueue;
import java.util.Comparator;
public class Solution {
PriorityQueue<Integer> highQueue = new PriorityQueue<>(); // 小顶堆保存大数据部分
PriorityQueue<Integer> lowQueue = new PriorityQueue<>((Comparator<Integer>) (o1, o2) -> o2 - o1);// 大顶堆保存小数据部分
public void Insert(Integer num) { // 保持两个堆平衡即size之差为1(默认lowQueue比highQueue >= 1)
highQueue.offer(num);// 数据先往大堆扔
lowQueue.offer(highQueue.poll());// 保证存到小堆的永远比大堆中的大。
if (lowQueue.size() > highQueue.size()) {// 若此时小堆容量大于大堆则移出小堆中最小数据至小堆
highQueue.offer(lowQueue.poll());
}
}
public Double GetMedian() {
return lowQueue.size() == highQueue.size() ? ((lowQueue.peek() + highQueue.peek()) * 0.5) : Double.valueOf(highQueue.peek());
}
}
这里的Double.valueOf值得关注一下,这里是不管你是字符 如 (‘3’)还是数字,统一会转化为Double的数。