数据流中的中位数 java

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

import java.util.ArrayList;
import java.util.PriorityQueue;

public class Solution {
    /**
     *如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，
     * 那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，
     * 那么中位数就是所有数值排序之后中间两个数的平均值。
     * 我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。
     *
     * 使用java中的优先队列，然后计算位置，取出数据，
     * 因为在取数据的过程中会把队列的数据取出，记录取出的数据，放回去就行了
     * 看有的人的挺复杂，可能是追求时间很短吧
     * 我这个我感觉时间复杂度NlogN
     * */
    PriorityQueue<Integer> p = new PriorityQueue<>();
    public void Insert(Integer num) {
        p.add(num);
    
    }

    public Double GetMedian() {
        int len = p.size();
        int count = (len + 1) / 2;
        ArrayList<Integer> lists = new ArrayList<>();
        for (int i = 0; i < count; ++i) {
            lists.add(p.poll());
        }
        if (len % 2 != 0) {
            p.addAll(lists);
            return lists.get(count - 1) * 1.0;
        } else {
            lists.add(p.poll());
            p.addAll(lists);
            return (lists.get(count - 1) + lists.get(count)) / 2.0;
        }
    }


}