数据流中的中位数(Java实现)

本题为剑指offer 面试题64

牛客网测试地址:https://www.nowcoder.com/questionTerminal/9be0172896bd43948f8a32fb954e1be1

[编程题]数据流中的中位数
  • 热度指数:24579  时间限制:1秒  空间限制:32768K
如何得到一个数据流中的中位数?如果从数据流中读出奇数个数值,那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值,那么中位数就是所有数值排序之后中间两个数的平均值。

Java Code:

package go.jacob.day614;

import java.util.Comparator;
import java.util.PriorityQueue;

import org.junit.Test;

public class Demo1 {
	private int count = 0;
	// 左半元素
	private PriorityQueue maxHeap = new PriorityQueue(15, new Comparator() {

		@Override
		public int compare(Integer o1, Integer o2) {
			return o2 - o1;
		}
	});
	// 右半元素
	private PriorityQueue minHeap = new PriorityQueue();

	public void Insert(Integer num) {
		count++;
		// 如果是偶数,插入到最小堆(右).
		// 必须先插入最大堆,然后把最大堆的根元素插入最小堆
		
		if (count % 2 == 0) {
			maxHeap.offer(num);
			int temp = maxHeap.poll();
			minHeap.offer(temp);
		}else{
			//如果是奇数,先放入最小堆,再把最小堆的根元素放入最大堆
			minHeap.offer(num);
			int temp=minHeap.poll();
			maxHeap.offer(temp);
		}
	}

	/*
	 * 如果数据总数目是偶数 . 返回最小堆根元素与最大堆根元素的均值 否则返回最大堆的根元素
	 */
	public Double GetMedian() {
		if (count % 2 == 0) {
			return new Double(maxHeap.peek() + minHeap.peek()) / 2;
		} else {
			return new Double(maxHeap.peek());
		}
	}

}


你可能感兴趣的:(剑指offer(Java实现))