PAT (Advanced Level) Practice_1057 Stack (30 分)_利用两个set的解法（小白解法）

• 题目地址
• 题目分析：这道题目，解法很多，值得考究，这里提供一种小白解法--利用两个set来模拟出平衡的节点左右两坨。
• 我的代码：

#include<iostream>
#include<set>
#include<stack>
using namespace std;

multiset<int>e1, e2;
stack<int>tt;
int med;

int balance(void);

int main()
{
	int n;
	scanf("%d", &n);

	char ss[11] = { '\0' };
	for (int i = 0, t; i < n; i++)
	{
		scanf("%s", ss);
		if (ss[1] == 'u')
		{
			scanf("%d", &t);
			tt.push(t);
			if (tt.size() == 1)med = t;

			if (t <= med)	e1.insert(t);
			else			e2.insert(t);

			med = balance();
		}
		else if (ss[1] == 'o')
		{
			if (tt.size())
			{
				if (tt.top() <= med)
					e1.erase(e1.find(tt.top()));
				else e2.erase(e2.find(tt.top()));
				med = balance();

				printf("%d\n", tt.top());
				tt.pop();
			}
			else puts("Invalid");
		}
		else
		{
			if (tt.size())
				printf("%d\n", med);
			else puts("Invalid");
		}
	}

	return 0;
}
int balance(void)
{
	if (e1.size() == e2.size() && e2.size() == 0) return 0;
	if (e1.size() - e2.size() == 2)//分析可知在每次平衡的前提下，仅仅4种情况+1，-1；+2，0
	{
		auto it = e1.end();
		it--;
		e2.insert(*it);
		e1.erase(it);
	}
	else if (e1.size() - e2.size() == -1)//+1，0是平衡的，故不处理
	{
		auto it = e2.begin();
		e1.insert(*it);
		e2.erase(it);
	}

	auto it = e1.end();
	it--;
	return (*it);
}