【Java】剑指Offer_编程题_数据流中的中位数

题目链接：https://www.nowcoder.com/questionTerminal/9be0172896bd43948f8a32fb954e1be1

题目描述
如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

第一次编辑代码：

import java.util.*;
public class Solution {
    int index = 0;
    int cap = 0;
    int [] ans = new int [100];
    public void Insert(Integer num) {
        ans[index++] = num;
        cap++;
    }

    public Double GetMedian() {
        Arrays.sort(ans);
        if(cap%2 == 0){
            return (double)(ans[cap/2] + ans[cap/2+1])/2;
        }else{
            return (double)ans[cap/2];
        }
    }
}

提交结果
答案错误:您提交的程序没有通过所有的测试用例

反思
这我是没想到的。

第二次编辑代码：

import java.util.*;
public class Solution {
    int cap = 0;
    PriorityQueue<Integer> low = new PriorityQueue<Integer>();
    PriorityQueue<Integer> high = new PriorityQueue<Integer>(new Comparator<Integer>(){
        public int compare(Integer o1,Integer o2){
            return o2.compareTo(o1);
        }
    });
    public void Insert(Integer num) {
        cap++;
        if(cap%2 != 0){
            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() {
        double ans;
        if(cap%2 != 0){
            ans = high.peek();
        }else{
            ans = (low.peek()+high.peek())/2.0;
        }
        return ans;
    }
}