输入:
17
Pop
PeekMedian
Push 3
PeekMedian
Push 2
PeekMedian
Push 1
PeekMedian
Pop
Pop
Push 5
Push 4
PeekMedian
Pop
Pop
Pop
Pop
输出:
Invalid
Invalid
3
2
2
1
2
4
4
5
3
Invalid
解题思路:
设置两个vector,一个v1用来模拟堆栈情况,一个v用来维护当前栈中元素的大小顺序。
对于push操作,v1只需要在后面加上一个元素即可,至于v,我们需要利用lower_bound()函数找到第一个大于等于当前元素的位置,然后将这个数字插入到相应位置。
对于pop操作,v1只需要将最后一个 元素弹出,v则是需要利用上述函数找到栈顶元素所在的位置,然后将这个元素从v中删除。
对于PeekMedian操作,首先判断当前v的大小的奇偶性,然后按照题目要求输出中间元素即可。
上代码:
#include <bits/stdc++.h>
using namespace std;
int n;
vector<int>v1,v;
int main()
{
cin>>n;
for(int i=1;i<=n;i++)
{
string str;
vector<int>::iterator it;
cin>>str;
if(str=="Push")
{
int x;
cin>>x;
v1.push_back(x);
it=lower_bound(v.begin(),v.end(),x);
v.insert(it,x);
}
else if(str=="Pop")
{
if(v1.size()==0)
cout<<"Invalid"<<endl;
else
{
it=lower_bound(v.begin(),v.end(),v1[v1.size()-1]);
v.erase(it);
cout<<v1[v1.size()-1]<<endl;
v1.pop_back();
}
}
else
{
if(v1.size()==0)
cout<<"Invalid"<<endl;
else
{
if(v.size()%2)
cout<<v[v1.size()/2]<<endl;
else
cout<<v[v1.size()/2-1]<<endl;
}
}
}
return 0;
}