数据流中的中位数

来源：互联网发布：python 搭建技术博客编辑：程序博客网时间：2024/06/10 03:15

从一个数据流中获取其中位数

/** * 维护两个优先队列,一个存储从中间到最大值的一半的数top , 这个队列为最小优先队列 * 另一个存储从中间到最小的一半数down, 这个队列为最大优先队列 */import java.util.Comparator;import java.util.PriorityQueue;import java.util.Queue;public class Solution {    private Queue<Integer> top = new PriorityQueue<Integer>(11 , new Comparator<Integer>() {@Overridepublic int compare(Integer o1, Integer o2) {if(o1 > o2 )return 1 ; else if(o1 < o2) return -1 ; else return 0; }}) ; private Queue<Integer> down = new PriorityQueue<Integer>(11 , new Comparator<Integer>(){@Overridepublic int compare(Integer o1, Integer o2) {if(o1 > o2)return -1 ; else if(o1 < o2)return 1 ; else return 0;}}) ;     public void Insert(Integer num) {        if(top.size() == down.size()){        if((!down.isEmpty() && num >= down.peek())||down.isEmpty()){        top.add(num) ;             }else{        top.add(down.poll()) ;         down.add(num) ;         }        }else{        if(num <= top.peek()){        down.add(num) ;         }else{        down.add(top.poll()) ;         top.add(num) ;         }        }    }    public Double GetMedian() {    if(top.size() > down.size())    return (double)top.peek() ;     return (top.peek() + down.peek())/2.0 ;     }}

0 0