算法--用大堆顶小堆顶结合求中间数

最新推荐文章于 2024-08-14 12:59:12 发布

tangyuan_sibal

最新推荐文章于 2024-08-14 12:59:12 发布

阅读量514

点赞数

分类专栏：算法

本文链接：https://blog.csdn.net/tangyuan_sibal/article/details/88805093

版权

算法专栏收录该内容

37 篇文章 2 订阅

订阅专栏

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

思路：大堆顶和小堆顶的特点就是最大值在最上面和最小值在最上面，如果在小堆顶的顶部存比大堆顶的顶部大的数，则整个小堆顶一定都比大堆顶的数大，这样就分为了两部分，而且中间数刚好在两个堆顶的顶部.（下面为某大佬的解题思路）

链接：https://www.nowcoder.com/questionTerminal/9be0172896bd43948f8a32fb954e1be1
来源：牛客网
先用java集合PriorityQueue来设置一个小顶堆和大顶堆
主要的思想是：因为要求的是中位数，那么这两个堆，大顶堆用来存较小的数，从大到小排列；
小顶堆存较大的数，从小到大的顺序排序*，显然中位数就是大顶堆的根节点与小顶堆的根节点和的平均数。
⭐保证：小顶堆中的元素都大于等于大顶堆中的元素，所以每次塞值，并不是直接塞进去，而是从另一个堆中poll出一个最大（最小）的塞值
⭐当数目为偶数的时候，将这个值插入大顶堆中，再将大顶堆中根节点（即最大值）插入到小顶堆中；
⭐当数目为奇数的时候，将这个值插入小顶堆中，再讲小顶堆中根节点（即最小值）插入到大顶堆中；
⭐取中位数的时候，如果当前个数为偶数，显然是取小顶堆和大顶堆根结点的平均值；如果当前个数为奇数，显然是取小顶堆的根节点
理解了上面所述的主体思想，下面举个例子辅助验证一下。
例如，传入的数据为：[5,2,3,4,1,6,7,0,8],那么按照要求，输出是"5.00 3.50 3.00 3.50 3.00 3.50 4.00 3.50 4.00 "
那么整个程序的执行流程应该是（用min表示小顶堆，max表示大顶堆）：
5先进入大顶堆，然后将大顶堆中最大值放入小顶堆中，此时min=[5],max=[无]，avg=[5.00]
2先进入小顶堆，然后将小顶堆中最小值放入大顶堆中，此时min=[5],max=[2],avg=[(5+2)/2]=[3.50]
3先进入大顶堆，然后将大顶堆中最大值放入小顶堆中，此时min=[3,5],max=[2],avg=[3.00]
4先进入小顶堆，然后将小顶堆中最小值放入大顶堆中，此时min=[4,5],max=[3,2],avg=[(4+3)/2]=[3.50]
1先进入大顶堆，然后将大顶堆中最大值放入小顶堆中，此时min=[3,4,5],max=[2,1]，avg=[3/00]
6先进入小顶堆，然后将小顶堆中最小值放入大顶堆中，此时min=[4,5,6],max=[3,2,1],avg=[(4+3)/2]=[3.50]
7先进入大顶堆，然后将大顶堆中最大值放入小顶堆中，此时min=[4,5,6,7],max=[3,2,1],avg=[4]=[4.00]
0先进入小顶堆，然后将小顶堆中最大值放入小顶堆中，此时min=[4,5,6,7],max=[3,2,1,0],avg=[(4+3)/2]=[3.50]
8先进入大顶堆，然后将大顶堆中最小值放入大顶堆中，此时min=[4,5,6,7,8],max=[3,2,1,0],avg=[4.00]

import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
	//判断是偶数插入还是奇数插入
	int count=0;
	//1、构建小顶堆，大顶堆,PriorityQueue默认是小顶堆
	private PriorityQueue<Integer> minHeap = new PriorityQueue<>();
	private PriorityQueue<Integer> maxHeap = new PriorityQueue<>(15,new Comparator<Integer>() {

				@Override
				public int compare(Integer o1, Integer o2) {
					// TODO Auto-generated method stub
					return o2-o1;
				}
			});
	public void Insert(Integer num)
	{
        ++count;
		//2、首先进大顶堆，然后将最大值插入小顶堆，当是第偶数个时
		//插入小顶堆，然后将最小值移至大顶堆
		
		//第偶数个插入到小顶堆，然后将小顶堆的最小值插到大顶堆
		if((count%2)==0)
		{
			minHeap.offer(num);
			Integer min = minHeap.poll();
			maxHeap.offer(min);
			
		}
		//第奇数个插入到大顶堆，然后将大顶堆的最大值插入到小顶堆
		else
		{

            maxHeap.offer(num);
			Integer max = maxHeap.poll();
			minHeap.offer(max);
			
		}

	}
	public Double GetMedian(){
		
		double result = 0.0; 
        if(count==0)
            throw new RuntimeException("no available number!");
		if((count%2)==0)
		{
			return (minHeap.peek()+maxHeap.peek())/2.0;
		}
		else
		{
			return result = minHeap.peek();
		}
		
	}




}