算法题（二十三）：利用堆解决流数据的中位数问题

最新推荐文章于 2024-08-14 09:00:00 发布

另一个我竟然存在

最新推荐文章于 2024-08-14 09:00:00 发布

阅读量837

点赞数

分类专栏：算法编程练手

本文链接：https://blog.csdn.net/qq_24034545/article/details/83625357

版权

该博客讨论了如何利用大顶堆和小顶堆解决流数据中位数问题。当数据量为奇数时，中位数是小顶堆的堆顶；为偶数时，是两大顶堆堆顶元素的平均值。通过Insert()方法插入数据，GetMedian()方法获取中位数。

摘要由CSDN通过智能技术生成

题目描述

如何得到一个数据流中的中位数？如果从数据流中读出奇数个数值，那么中位数就是所有数值排序之后位于中间的数值。如果从数据流中读出偶数个数值，那么中位数就是所有数值排序之后中间两个数的平均值。我们使用Insert()方法读取数据流，使用GetMedian()方法获取当前读取数据的中位数。

分析

利用一个大顶堆和小顶堆来解决问题。当输入数据的总量为奇数时，数据先放入小顶堆，再从小顶堆拿出堆顶元素放入大顶堆中；当为偶数时，数据先放入大顶堆，再从大顶堆中拿出堆顶元素放入小顶堆中。保证小顶堆的最小值比大顶堆的最大值要大。输出数据时，当总量为奇数时，直接输出小顶堆的堆顶元素，当为偶数时，输出大顶堆和小顶堆的堆顶元素的平均值。

代码

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

public class StreamMedian {

	private int count = 0;
	private PriorityQueue<Integer> minHeap = new PriorityQueue<>();
	private PriorityQueue<Integer> maxHeap = new PriorityQueue<>(15,new Comparator<Integer>(){

		@O