描述
比如:{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); } }