中位数【洛谷P1168】

题目描述

给出一个长度为N的非负整数序列A[i]，对于所有1 ≤ k ≤ (N + 1) / 2，输出A[1], A[3], …, A[2k - 1]的中位数。即前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]的中位数。
输入样例
7
1 3 5 7 9 11 6
输出样例
1
3
5
6
说明
对于20%的数据，N ≤ 100；
对于40%的数据，N ≤ 3000；
对于100%的数据，N ≤ 100000.

题解：
建一个小根堆，一个大根堆，以根为mid值，大数扔进小根堆，小数扔进大根堆，保证两个堆元素总量相等即可，根就是题目所求中位数
code

#include<queue>
#include<cstdio>
#include<iostream>
#include<algorithm>
using namespace std;
int a[100005];
priority_queue<int>q1;
priority_queue<int,vector<int>,greater<int> >q2;
int main()
{
	int n,mid,i;
    scanf("%d",&n);
    scanf("%d",&a[1]);
    mid=a[1];
    printf("%d\n",mid);
    for(i=2;i<=n;i++)
	{
        scanf("%d",&a[i]);
        if(a[i]>mid) 
			q2.push(a[i]);//比中间数大，入小根堆 
        else 
			q1.push(a[i]);//否则，入大根堆 
        if(i%2)//奇数，调整两个堆的大小 
		{
            while(q1.size()!=q2.size())
			{
                if(q1.size()>q2.size())
				{
                    q2.push(mid);
                    mid=q1.top();
                    q1.pop();
                }
                else
				{
                    q1.push(mid);
                    mid=q2.top();
                    q2.pop();
                }
            }
           	printf("%d\n",mid); 
        }
    }
    return 0;
}