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

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

multisete1, e2;
stacktt;
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);
}

 

你可能感兴趣的:(PAT,(Advanced,Level),Practice)