洛谷P1168中位数

P1168中位数

题目分析：

• 本题意思是再输入n个序列成员后，依次输出几位个时的序列中位数

这题其实有很多的解题方法，但最近在学STL就想用其中的技巧来解题
可以使用vetcor来模拟序列容器。在每次输入时都进行一次排序，让序列一直保持有序状态，当第奇数次输入时，输出当时序列的奇数位就行。

#include <bits/stdc++.h>//导入头函数
using namespace std;
int n;
vector<int>v;//定义vector容器
int main()
{
    cin >> n;//输入序列长度
    for (int i = 1, x; i <= n; i++) //从1开始有利于后面的中位数存放
    {
        scanf("%d", &x);
        //upper_bound函数直接找到大于输入值的位置，insert将输入值插入
        v.insert(upper_bound(v.begin(), v.end(), x), x);
        if (i % 2 == 1) //当存放次数为奇数时
        {
            printf("%d\n", v[(i - 1) / 2]); //输出中位数的值
        }
    }
    return 0;
}

但除此之外我还了解到有人用优先队列当做容器，我一开始认为优先队列无法取队中的元素所以一直认为这不可能，但后来知道有个平衡堆的方法，确实让我很震惊，确实是自己格局小了。

#include<bits/stdc++.h>//导入头文件
#include<queue>//导入队列库
using namespace std;
int n;
int a[100100];
int mid;
priority_queue<int,vector<int>,less<int> >q1;//大根堆，逆排
priority_queue<int,vector<int>,greater<int> >q2;//小根堆，顺排
int main(){
    cin>>n;
    scanf("%d",&a[1]);
    mid=a[1];
    cout<<mid<<endl;//mid初值是a[1]
    for(int 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==1){//第奇数次加入
            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();
                }
            }
            cout<<mid<<endl;
        }
    }
    return 0;
}

这个代码逻辑清晰，并且用到了优先队列的有序性，以及一些不错的算法思想，值得学习思考。