洛谷 中位数

点击就送屠龙宝刀

题目描述 Description
给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[2], …, A[2k - 1]的中位数。即前1，3，5，……个数的中位数。
输入输出格式 Input/output
输入格式：
输入文件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]的中位数。
输入输出样例 Sample input/output
输入样例
7
1 3 5 7 9 11 6
输出样例
1
3
5
6
数据范围
对于20%的数据，N ≤ 100；
对于40%的数据，N ≤ 3000；
对于100%的数据，N ≤ 100000。
这道题洛谷给的定位是普及-提高然而我觉得这TM就是个脑洞题目。。首先暴力的做法肯定都会。读进来一个数sort一下输出(注意不是真的读进来一个就sort一下)。当然也有优化的地方但是显然数据不会很水。。然后我们想到set是自动排序的且复杂度很可观，那么我们只需要用lower_bound查询中位数，然后输出就好了。但是很可惜只能水过40%的数据。继续想还有什么办法呢？对于自动排序的数据结构自然C艹党想到了堆。没错。我们开两个堆，对于这两个堆a,b，一个用来维护输入的数字，一个记录比中位数大的数字，然后就可以做到O2查询。为什么是O2呢因为你要判断中位数在不在第一个堆里。代码很简单但是对于长期撸模板题的同学来说还是有一定思维难度的。

#include<iostream>
#include <cstdio>
#include <queue>

using namespace std;

int main()
{
    int n,a;
    scanf("%d",&n);
    priority_queue <int> max,min;
    bool odd=true;
    scanf("%d",&a);
    printf("%d\n",a);
    max.push(-a);
    for (int i=1;i<n;i++)
    {
        odd=odd-1;
        scanf("%d",&a);
        if (a>=-max.top()) max.push(-a); else min.push(a);
        if (max.size()-min.size()==2) {min.push(-max.top());max.pop();}
            else
            if (min.size()-max.size()==2) {max.push(-min.top());min.pop();}
        if (odd)
            if (max.size()>min.size()) printf("%d\n",-max.top());
            else printf("%d\n",min.top());
    }
    return 0;
}