AcWing 106 动态中位数

题目描述:

依次读入一个整数序列,每当已经读入的整数个数为奇数时,输出已读入的整数构成的序列的中位数。

输入格式

第一行输入一个整数P,代表后面数据集的个数,接下来若干行输入各个数据集。每个数据集的第一行首先输入一个代表数据集的编号的整数。然后输入一个整数M,代表数据集中包含数据的个数,M一定为奇数,数据之间用空格隔开。

数据集的剩余行由数据集的数据构成,每行包含10个数据,最后一行数据量可能少于10个,数据之间用空格隔开。

输出格式

对于每个数据集,第一行输出两个整数,分别代表数据集的编号以及输出中位数的个数(应为数据个数加一的二分之一),数据之间用空格隔开。数据集的剩余行由输出的中位数构成,每行包含10个数据,最后一行数据量可能少于10个,数据之间用空格隔开。输出中不应该存在空行。

数据范围

1≤P≤1000,
1≤M≤9999

输入样例:

3 
1 9 
1 2 3 4 5 6 7 8 9 
2 9 
9 8 7 6 5 4 3 2 1 
3 23 
23 41 13 22 -3 24 -31 -11 -8 -7 
3 5 103 211 -311 -45 -67 -73 -81 -99 
-33 24 56

输出样例:

1 5
1 2 3 4 5
2 5
9 8 7 6 5
3 12
23 23 22 22 13 3 5 5 3 -3 
-7 -3

分析:

本题与AcWing 54 数据流中的中位数基本相同。

维护一个对顶堆,即一个大根堆和一个小根堆,大根堆里存储较小的一半数据,小根堆里存储较大的一半数据。读入数据先存入大根堆,如果小根堆不为空且小根堆堆顶小于大根堆堆顶则交换两个堆顶元素,如果小根堆元素个数比大根堆元素个数少了2,则将大根堆堆顶元素插入到小根堆中。每次查询中位数输出大根堆堆顶即可。

#include <iostream>
#include <cstring>
#include <queue>
using namespace std;
const int maxn = 10005;
int p,m,n,a;
int main(){
	cin>>p;
	while(p--){
		cin>>m>>n;
		cout<<m<<" "<<(n + 1) / 2<<endl;
		priority_queue<int> ma;
		priority_queue<int,vector<int>,greater<int> > mi;
		for(int i = 1;i <= n;i++){
			cin>>a;
			ma.push(a);
			if(mi.size() && mi.top() < ma.top()){
				int x = ma.top(),y = mi.top();
				ma.pop(),mi.pop();
				ma.push(y),mi.push(x);
			}
			if(ma.size() == mi.size() + 2){
				int x = ma.top();
				mi.push(x);
				ma.pop();
			}
			if(i % 2 == 1)	cout<<ma.top()<<" ";
			if((i + 1) % 20 == 0)	cout<<endl;
		}
		cout<<endl;
	}
	return 0;
}

 

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值