题目:
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。
思路:
我们选用存放数据的容器是最大堆和最小堆,左侧为最大堆,右侧为最小堆。最大堆的数字全部小于最小堆的数字,且最小堆的尺寸要么与最大堆相同,要么比最大堆尺寸大一。因此,当原本两堆大小相同时,新来的数字首先被加到最大堆,之后最大堆将堆顶数字加给最小堆(先加入最大堆的意义在于,保证传递给最小堆数字后,仍保持两堆大小关系),此时的中位数就是最小堆的堆顶。当最小堆尺寸比最大堆大一时,原步骤不变,之后因为最大堆尺寸比最小堆小2,因此最小堆再将堆顶传给最大堆。此时中位数是两堆和的一半。
代码比较明了
import java.util.*;
//PriorityQueue默认是最小堆,设置最大堆时要重新设置比较器
public class Solution {
public static PriorityQueue<Integer> minheap, maxheap;
public Solution(){
minheap = new PriorityQueue<>();
maxheap = new PriorityQueue<>(20, new Comparator<Integer>() {
@Override
public int compare(Integer o1, Integer o2) {
return o2-o1;
}
});
}
public static void addnum(int n){
//这里最小堆没有直接加新输入的值,而是先把数值添加到最大堆,再添加最大堆堆顶。保证最小堆新添加的值大于等于最大堆所有值
maxheap.offer(n);
minheap.offer(maxheap.poll());
if (maxheap.size()+1<minheap.size())
{
maxheap.offer(minheap.poll());
}
}
public static double GetMedian()
{
if (maxheap.size()==0&&minheap.size()==0)
return -1;
if (maxheap.size()==0)
return minheap.peek();
if (minheap.size()==maxheap.size())
return (minheap.peek()+maxheap.peek())*0.5;
else return minheap.peek();
}
}