PAT 1057 Stack

最新推荐文章于 2023-01-29 15:23:30 发布

原创最新推荐文章于 2023-01-29 15:23:30 发布 · 172 阅读

0 ·

CC 4.0 BY-SA版权

算法竞赛专栏收录该内容

1 篇文章

订阅专栏

本文介绍了一种使用两个多集合维护一系列整数中位数的算法。通过不断调整两个按不同顺序排序的多集合（一个升序，一个降序），确保在任何时候都能快速找到当前整数序列的中位数。该算法支持插入新数值和删除已存在数值的操作。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

#include <cstdio>
#include <set>
#include <cstring>
#include <stack>
using namespace std;

int main()
{
	int i, j, k, t, d;
	stack<int>s;
	char buff[12];
	multiset<int>upper;   //从小到大排序 
	multiset<int, greater<int> >lower;//从大到小排序 
	
	//我把中位数放在了大的集合里，所以upper里的元素数量比lower的多一个或两个
	//如果数量一样就是没有一个数字 
	
	scanf("%d",&t);
	while(t--)
	{
		scanf("%s",buff);
		if(buff[1] == 'o')
		{
			if(s.empty())
			{
				printf("Invalid\n");
			}
			else
			{
				d = s.top();
				printf("%d\n",d);
				s.pop();
				if(d >= *upper.begin())
				{
					upper.erase(upper.find(d));// 放入大的集合 
				}
				else
				{
					lower.erase(lower.find(d));//放入小的集合 
				}
				if(upper.size() == lower.size() && upper.size() > 0)///保证 upper.size() > 0
				{
					upper.insert(*lower.begin());
					lower.erase(lower.begin());
				}
				else if(upper.size() - lower.size() > 2)
				{
					lower.insert(*upper.begin());
					upper.erase(upper.begin());
				}
			}
		}
		else if(buff[1] == 'u')
		{
			scanf("%d",&k);
			s.push(k);
			if(lower.size() && k < *lower.begin())//保证lower.size()>0
			{
				lower.insert(k);
			}
			else
			{
				upper.insert(k);
			}
			if(upper.size() == lower.size() && upper.size() > 0)
			{
				upper.insert(*lower.begin());
				lower.erase(lower.begin());
			}
			else if(upper.size() - lower.size() > 2)
			{
				lower.insert(*upper.begin());
				upper.erase(upper.begin());
			}	
		}
		else 
		{
			if(upper.size())
				printf("%d\n",*upper.begin());
			else printf("Invalid\n");
		}
	}
	
	return 0;
}