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;
}
这个代码逻辑清晰,并且用到了优先队列的有序性,以及一些不错的算法思想,值得学习思考。