无序数组的中位数

描述

比如:{1,3,2,5,4} 中位数为3

{1,3,2,5,4,6}中位数为3.5

解析

排序

当然 排序啊。快排。

最小堆

首先将数组的前(n+1)/2个元素建立一个最小堆。

然后,对于下一个元素,和堆顶的元素比较,如果小于等于,丢弃之,接着看下一个元素。如果大于,则用该元素取代堆顶,再调整堆,接着看下一个元素。重复这个步骤,直到数组为空。

当数组都遍历完了,那么,堆顶的元素即是中位数。

可以看出,长度为(n+1)/2的最小堆是解决方案的精华之处。

代码

最小堆

public static Double mm(int[] n) {
        if (n == null || n.length <= 0) {
            return null;
        }
        int arraySize = n.length;
        int limitSize = ((arraySize) >> 1) + 1;
        PriorityQueue<Integer> queue = new PriorityQueue<>(limitSize);
        for (int i = 0; i < limitSize; i++) {
            queue.offer(n[i]);
        }
        for (int i = limitSize; i < arraySize; i++) {
            if (n[i] > queue.peek()) {
                queue.poll();
                queue.offer(n[i]);
            }
        }
        if ((arraySize & 1) == 1) {
            return (double)queue.peek();
        } else {
            return (double)((queue.poll() + queue.peek()) >> 1);
        }
    }

猜你喜欢

转载自www.cnblogs.com/fanguangdexiaoyuer/p/10622378.html