HDU5249(set+queue模拟)

最新推荐文章于 2020-02-23 10:18:49 发布

原创最新推荐文章于 2020-02-23 10:18:49 发布 · 456 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#C++ #算法 #模拟 #set #queue

算法专栏收录该内容

48 篇文章

订阅专栏

本文介绍了一种使用set和queue实现动态查询中位数的方法。通过插入元素并保持两个集合平衡来确保快速查找中位数，同时支持元素删除。

这种动态查询中位数的题目比较老了。。。我在这篇博客的最后面也讲到了点击我

原理是一样的，只不过把堆改成了set，因为要删除元素再加一个queue记录进入顺序。

我这里用比较简练的方式实现了。

        #include<iostream>
        #include<set>
        #include<queue>
        #include<algorithm>
        #include<cstdio>
        using namespace std;
		
		int main()
{
	    int Case=1;
	    int n;
	    while(cin>>n)
	    {
	    	printf("Case #%d:\n",Case++);
	    	char s[20];
	    	queue<int> q;
	    	set<int> left;
	    	set<int> right;
	    	int total=0;
	    	for(int i=1;i<=n;i++)
	    	{
	    		scanf("%s",s);
	    	    if(s[0]=='i')
	    	    {
	    	    	int num;
	    	    	scanf("%d",&num);
	    	    	q.push(num);
	    		    if(total==0||num<*left.rbegin())
	    		    	left.insert(num);
					else
					    right.insert(num);
			        total++;
			    }
			    else if(s[0]=='o')
			    {
			    	int num=q.front();
			    	q.pop();
			    	total--;
			    	if(left.find(num)!=left.end())
			    		left.erase(num);
					else
						right.erase(num);
				}
				else
					printf("%d\n",*left.rbegin());
				
				//调整左右高度	
				if(total!=0&&total%2==0&&left.size()==right.size())
				{
					left.insert(*right.begin());
					right.erase(*right.begin());
				}
				else if(total%2==1&&left.size()-right.size()>2)
				{
					right.insert(*left.rbegin());
					left.erase(*left.rbegin());
				}
		    }
		}
}