洛谷 P1168 中位数

题目描述

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[3], …, 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[3], …, 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。

两个堆 一大一小

大根堆存比中位数小的树，小根堆维护比中位数大的数。

屠龙宝刀点击就送

#include <algorithm>

#include <cstdio>

#include <queue>

using namespace std;

priority_queue<int>gr;

priority_queue<int,vector<int>,greater<int> >les;

int syg,n,now;

int main()

{

    scanf("%d",&n);

    scanf("%d",&now);

    printf("%d\n",now);les.push(now);

    syg=now;

    for(int x,y,i=;i<=(n-)/;++i)

    {

        scanf("%d%d",&x,&y);

        if(syg>x) gr.push(x);else les.push(x);

        if(syg>y) gr.push(y);else les.push(y);

        while(gr.size()>=les.size()) {les.push(gr.top());gr.pop();}

        while(gr.size()<les.size()-){gr.push(les.top());les.pop();}

        printf("%d\n",les.top());

        syg=les.top();

    }

    return ;

}