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;
}

 

posted @ 2016-12-24 20:52  IIIIIIIIIU  阅读(223)  评论(0编辑  收藏  举报