无序数组取中位数

最暴力的 - 先排序

快一点的 - 快排,任选一个数,小的放前面,大的放后面,如果左部分的数量正好是(n-1)/ 2 那支点就找到了。否则的话,就继续找。

优雅的 - 构建一个最小堆,用前(n+1)/ 2 个元素构建一个最小堆,将其余元素尝试向里插入,比堆顶小的扔了,比堆顶大的留下,最后的队形就是中位数了。

public static double median(int[] array){
    int heapSize = array.length/2 + 1;
    PriorityQueue<Integer> heap = new PriorityQueue<>(heapSize);
    for(int i=0; i<heapSize; i++){
        heap.add(array[i]);
    }
    for(int i=heapSize; i<array.length; i++){
        if(heap.peek()<array[i]){
            heap.poll();
            heap.add(array[i]);
        }
    }
    if(array.length % 2 == 1){
        return (double)heap.peek();
    }
    else{
        return (double)(heap.poll()+heap.peek())/2.0;
    }
}

参考: https://www.cnblogs.com/shizhh/p/5746151.html

猜你喜欢

转载自www.cnblogs.com/Fivee/p/12915922.html