L3-002 特殊堆栈

输入：

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