【题解】中位数

8 篇文章 0 订阅
3 篇文章 0 订阅

题目来源:洛谷
在这里插入图片描述

思路:

维护两个堆,一个是大根堆,,一个是小根堆,小根堆的堆顶是较大的数中最小的,大根堆的堆顶是较小的数中最大的,这样一来答案就是小根堆的堆顶,因为小根堆内的元素个数要不就是等于大根堆的元素个数(当i为偶数时),要不就是比大根堆的元素个数多1(当i为奇数时)

code:
#include<bits/stdc++.h>
using namespace std;
int n;
priority_queue <int> q1;//大根堆 
priority_queue <int ,vector<int>,greater<int> > q2;//小根堆 
int main()
{
	scanf("%d",&n);
	for (int i=1;i<=n;i++)
	{
		int a;
		scanf("%d",&a);
		if (i==1) //特判 
		{
			q1.push(a);
			printf("%d\n",a);
			continue;
		}
		if (a<=q1.top()) q1.push(a);
		else q2.push(a); 
		 /*q1小根堆的堆顶是较大的数中最小的,q2大根堆的堆顶是较小的数中最大的
		   所以当a小于等于q1的堆顶时,也就是a比q1中所有的数都小,就放进q1中
		   否则a比q2的堆顶大,放进q2 
		 */
		while (q2.size()>q1.size()) 
		{
			q1.push(q2.top());
			q2.pop();
		}
		while (q1.size()>q2.size()+1) 
		//q2.size()+1是因为当i为奇数的时候q1的元素个数总是要比q2的元素个数大1 
		{
			q2.push(q1.top());
			q1.pop();
		} 
		if (i%2!=0) printf("%d\n",q1.top()); //i为奇数的时候q1的堆顶就是答案 
	}
	return 0;
}
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值