POJ 3784 最大堆最小堆求动态中位数

题意:读入一些32位int范围内的整数,当读入奇数个时,输出目前读入的这些值的中位数。
第一行一个整数P,表示测试数据组数。 每组测试数据,第一行一个整数n,表示元素个数。 接下里n个整数,每10个一行,最后一行可能不满10个。
输出中位数时,也需要10个一行,最后一行可能不满10个,但至少会有1个。
思路:
建两个堆, 大顶堆和小顶堆
每次输入一个数,如果这个数比当前的中位数大,就存入小顶堆中, 否则就存入大顶堆。
然后调整, 小顶堆元素的个数要等于大顶堆的元素个数,或者比其多1。
如果小顶堆的元素太多,就塞到大顶堆里,反之亦然
这样一来就会发现。小顶堆的元素比所有大顶堆的元素都大, 而且小顶堆的堆顶就是中位数。
思路借鉴于https://blog.csdn.net/sdj222555/article/details/43058443 写的很好!

#include<iostream>
#include<cstdio>
#include<map>
#include<algorithm>
#include<queue>
#define ll long long
using namespace std;
priority_queue<int,vector<int>,greater<int> > qx;
priority_queue<int,vector<int>,less<int> > qd;
void add(int x)
{
	if(qx.empty())
	{
		qx.push(x);
		return;
	}
	if(x>qx.top())
		qx.push(x);
	else qd.push(x);
	while(qx.size()<qd.size())
	{
		int temp=qd.top();
		qd.pop();
		qx.push(temp);
	}
	while(qx.size()>qd.size()+1)
	{
		int temp=qx.top();
		qx.pop();
		qd.push(temp);
	}
}
int main()
{
	int t,tt,n,i;
	cin>>t;
	while(t--)
	4{
		cin>>tt>>n;
		int a[10000],ans[10000],cnt=0,temp;
		while(!qx.empty())
			qx.pop();
		while(!qd.empty())
			qd.pop();
		for(i=1;i<=n;i++)
		{
			cin>>temp;
			add(temp);
			if(i%2)
				ans[cnt++]=qx.top();
		}
		cout<<tt<<" "<<cnt<<endl;
		for(i=0;i<cnt;i++)
		{
			cout<<ans[i]<<" ";
			if((i+1)%10==0)
				cout<<endl;
		}
		if(cnt%10)
			cout<<endl;
	}
	return 0;
} 

  • 1
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值