41、数据流中的中位数

数据流中的中位数

题目描述

如何得到一个数据流中的中位数？如果从数据流中读出奇数的个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序后中间两个数的平均值。

思路分析

用一个最大堆实现左边的数据容器，用一个最小堆实现右边的数据容器。首先要保证数据平均分配到两个堆中，因此两个堆中数据的数目之差不能小于1.所以在数据的总数目是偶数时将新数据插入最小堆，否则插入最大堆。代码如下：

import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
	PriorityQueue<Integer> minHeap=new PriorityQueue<>();
    PriorityQueue<Integer> maxHeap=new PriorityQueue<>(20,new Comparator<Integer>(){
         @Override
         public int compare(Integer o1, Integer o2) {
             return o2-o1;
         }
    });
    private int count=0;
    public void Insert(Integer num) {
    	count++;
        if(count%2==0){
            if(!maxHeap.isEmpty()&&maxHeap.peek()>num){
                maxHeap.add(num);
                num=maxHeap.poll();
            }
            minHeap.add(num);
        }else{
            if(!minHeap.isEmpty()&&minHeap.peek()<num){
                minHeap.add(num);
                num = minHeap.poll();
            }
            maxHeap.add(num);
        }
    }

    public Double GetMedian() {
        if(maxHeap.size()==minHeap.size()){
            return (maxHeap.peek()+minHeap.peek())/2.0;
        }else if(maxHeap.size()>minHeap.size()){
            return maxHeap.peek()*1.0;
        }else {
            return minHeap.peek()*1.0;
        }
    }
}