AC日记——中位数 洛谷 P1168

题目描述

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[2], …, A[2k - 1]的中位数。[color=red]即[/color]前1，3，5，……个数的中位数。

输入输出格式

输入格式：

输入文件median.in的第1行为一个正整数N，表示了序列长度。

第2行包含N个非负整数A[i] (A[i] ≤ 10^9)。

输出格式：

输出文件median.out包含(N + 1) / 2行，第i行为A[1], A[2], …, A[2i – 1]的中位数。

输入输出样例

输入样例#1：

7
1 3 5 7 9 11 6

输出样例#1：

1
3
5
6

说明

对于20%的数据，N ≤ 100；

对于40%的数据，N ≤ 3000；

对于100%的数据，N ≤ 100000。

思路：

　　两个堆，一个大根堆，一个小根堆

　　大根堆存比中位数小的数

　　小根堆存比中位数大的数

　　每读两个数就维护一下当前中位数

　　然后输出

　　轻松ac

来，上代码：

#include <queue>
#include <cstdio>

using namespace std;

int n,now,now_,size_s,size_l;

priority_queue<int>small;
priority_queue<int>large;

int main()
{
scanf("%d",&n);
scanf("%d",&now);
printf("%d\n",now);
for(int i=2;i<=n;i++)
{
scanf("%d",&now_);
if(now_>=now)
{
large.push(now_*-1);
size_l++;
}
else
{
small.push(now_);
size_s++;
}
if(i%2)
{
if(size_l>size_s)
{
now_=large.top()*-1;
small.push(now);
now=now_;
large.pop();
size_l--;
size_s++;
}
else if(size_s!=size_l)
{
now_=small.top();
large.push(now*-1);
now=now_;
small.pop();
size_s--;
size_l++;
}
printf("%d\n",now);
}
}
return 0;
}