PTA L3-002 特殊堆栈（STL模拟）

题意：

模拟堆栈，有三种操作Push，Pop，PeekMedian，其中“取中值”操作是取元素大小排名中间的元素。若操作非法，则对应输出 Invalid。

输入：

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，stack维持进出的顺序，vector维持元素的顺序，stack和vector中的元素要同步更新，但如果每次取中间值都用一次排序，复杂度就太高了，所以可以用二分来优化一下，在push stack同步插入vector的时候，按序插入，在pop stack同步删除vector的时候，按序删除，所以vector一直是有序排列的；但是我又想了为什么不能单独用一个vector来二分不是更简单了么？在线打脸，写了后面忘了前面，因为栈是元素顺序是vector的一个参照系。

#include<iostream>
#include<algorithm>
#include<cstring>
#include<vector>
using namespace std;
vector<int> stk, vec;
vector<int>::iterator it;

int main()
{
	int n, num;
	string s;
	cin >> n;
	for (int i = 0; i < n; i++)
	{
		cin >> s;
		if (s == "Pop")
		{
			if (stk.size() == 0)
				cout << "Invalid" <<endl;
			else
			{
				cout << stk[stk.size() - 1] << endl;
				it = lower_bound(vec.begin(), vec.end(), stk[stk.size() - 1]);
				vec.erase(it);
				stk.pop_back();
			}
		}
		else if (s == "PeekMedian")
		{
			int size = stk.size();
			if (size == 0)
				cout << "Invalid" <<endl;
			else // 在 vec 中找中间数
			{
				if (size & 1) // odd
					cout << vec[size/2] << endl;
				else
					cout << vec[size/2 - 1] << endl;
		    } 
		}
		else
		{
			cin >> num;	
			it = lower_bound(vec.begin(), vec.end(), num);
			vec.insert(it, num);
			stk.push_back(num);
		}
	}
	return 0;
}