如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流,使用GetMedian()方法获取当前读取数据的中位数。
解题思路:使用最大堆和最小堆,如果插入个数是奇数,则是中间的数,如果是偶数,则是中间两数之和的一半,比如1,2,3,4,5,6,那么中位数就是中间两位之和的一半,现在问题就转换为怎么定位这两个数,所以可以使用最大堆和最小堆,根据堆的性质,最大堆的堆顶元素最大,最小堆的堆顶元素最小,最大堆用于保存前半部分的数(1,2,3),最小堆用于保存后半部分的数(4,5,6),当插入的个数是奇数时,返回最大堆的堆顶即是中位数,如果是偶数,则是最大堆的堆顶和最小堆的堆顶之和的一半,;此时最大堆的堆顶元素为3,最小堆的堆顶为4,所以中位数为两者之和的一半(3+4)/ 2.0 = 3.5。
public class Solution {
private int cnt = 0;
private PriorityQueue<Integer> low = new PriorityQueue<>();
private PriorityQueue<Integer> high = new PriorityQueue<>( //默认是最小堆,最大堆需要重写方法
new Comparator<Integer>(){
public int compare(Integer o1, Integer o2){
return o2.compareTo(o1);
}
});
public void Insert(Integer num) {
++cnt;
if((cnt & 1) == 1){
if(!low.isEmpty() && num > low.peek()){ //当最小堆不为空,并且待插入的数大于最小堆堆顶时弹出堆顶
low.offer(num);
num = low.poll();
}
high.offer(num);
}
else{
if(!high.isEmpty() && num < high.peek()){ //当最大堆不为空,并且待插入的数小于最大堆堆顶时弹出
high.offer(num);
num = high.poll();
}
low.offer(num);
}
}
public Double GetMedian() {
if((cnt & 1) == 1){ //插入个数为奇数
return high.peek() * 1.0;
}else{ //插入个数为偶数
return (high.peek() + low.peek()) / 2.0;
}
}
}