题目:
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
思路:
建立两个堆,一个最小堆一个最大堆,要达到当元素数量是偶数时,中位数就是两个堆顶的平均值,如果为奇数,就是两个堆中多的那的堆的堆顶元素这样的效果。就需要在入堆时,
count
为偶数时,如果插入的数字小于大顶堆的堆顶,则该元素先入大顶堆,然后把大顶堆的堆顶放入小顶堆;否则,直接进入小顶堆。count
为奇数时,如果插入的数字大于小顶堆的堆顶,则该元素先入小顶堆,然后把小顶堆的堆顶放入大顶堆;否则,直接进入大顶堆。
代码:
private static PriorityQueue<Integer> min = new PriorityQueue<>();
private static PriorityQueue<Integer> max = new PriorityQueue<>(new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2 - o1;
}
});
static int count = 0;
private static void insert(int num) {
if ((count & 1) == 0) {
if (!max.isEmpty() && num < max.peek()) {
max.offer(num);
num = max.poll();
}
min.offer(num);
} else {
if (!min.isEmpty() && num > min.peek()) {
min.offer(num);
num = min.poll();
}
max.offer(num);
}
count++;
}
private static Double middleNum() {
if ((count & 1) == 0) {
return (double)(min.peek() + max.peek()) / 2;
} else {
return (double) min.peek();
}
}