动态中位数

106. 动态中位数(https://www.acwing.com/problem/content/108/)
依次读入一个整数序列,每当已经读入的整数个数为奇数时,输出已读入的整数构成的序列的中位数。

输入格式
第一行输入一个整数 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

解法一:
在不断输入数字的过程中,每次都有序的将新输入的数字插入序列,保证序列始终有序,中位数就在序列的中间位置,寻找插入的位置可以使用二分

AC代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const int inf = 0x3f3f3f3f;
int b[10000];
int main()
{
	int p;
	cin>>p;
	int a,n,cnt;
	vector<int>res;
	while (p--)
	{
		cnt=0;
		int f=0;
		cin>>a>>n;
		for(int i=1;i<=n;i++) cin>>b[i];
		cout<<a<<' '<<(n+1)/2<<endl;
		res.clear();
		res.push_back(-inf);
		for(int i=1;i<=n;i++){
			auto t=lower_bound(res.begin(),res.end(),b[i]);
			res.insert(t,b[i]);
			if(i&1){
				cnt++;
				if(cnt==10)
				{
					cout<<res[res.size()/2]<<endl;
					cnt=0;
				}
				else 
				cout<<res[res.size()/2]<<' ';
			}
			if(i==n&&cnt!=0&&cnt<10){
				f=1;
			}
		}
		if(f)
		cout<<endl;
	}
	
	return 0;
}

第二种解法:对顶堆
维护一个大根堆,一个小根堆,读入的过程中若比中位数大则插入小根堆,反之插入大根堆,当两堆元素数量不匹配时将数量多的堆堆顶插入数量少的堆中。(有STL为什么还要自己写一个堆来维护呢,对吧Q_Q)

AC代码:

#include<bits/stdc++.h>
using namespace std;
typedef long long int ll;
const int inf = 0x3f3f3f3f;
int main()
{
	int p;
	cin>>p;
	int a,b,n,cnt;
	while (p--)
	{
		cnt=0;
		int f=0;
		cin>>a>>n;
		priority_queue<int>big;
		priority_queue<int,vector<int>,greater<int>>sml;
		cout<<a<<' '<<(n+1)/2<<endl;
		int mid=-inf;
		for(int i=1;i<=n;i++){
			cin>>b;
			if(b>mid) sml.push(b);
			else big.push(b);

			if(big.size()>i/2){
				sml.push(big.top());
				big.pop();
			}
			if(sml.size()>(i+1)/2){
				big.push(sml.top());
				sml.pop();
			}
			if(i&1){
				cnt++;
				mid=sml.top();
				if(cnt==10)
				{
					cout<<mid<<endl;
					cnt=0;
				}
				else 
				cout<<mid<<' ';
			}
			if(i==n&&cnt!=0&&cnt<10){
				f=1;
			}
		}
		if(f)
		cout<<endl;
	}
	
	return 0;
}
  • 1
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值