hdu4006 The kth great number(优先队列）

最新推荐文章于 2019-09-12 19:00:38 发布

原创最新推荐文章于 2019-09-12 19:00:38 发布 · 335 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

搜索专栏收录该内容

6 篇文章

订阅专栏

本文介绍了一种利用优先队列高效求解前K大数的方法，并提供了完整的C++实现代码。该方法特别适用于需要频繁更新并快速获取当前最大的K个元素的场景。

题目链接：点击打开链接

据说有很多方法，不过如果想省事儿的话，
还是用优先队列吧，只存前k大的数就行了。

代码如下：

#include<iostream>
#include<queue>
#include<vector>
#include<functional>
using namespace std;

int main()
{
	int n,k,i;
	char s;
	int a;
	while(cin>>n>>k)
	{
	    priority_queue<int,vector<int>,greater<int> >q;
		i=0;
		while(n--)
		{
			cin>>s;
			if(s=='Q')
			cout<<q.top()<<endl;//输出第k大的数
			 else
			{
				cin>>a;
				if(i==k&&a>=q.top())//大于原来的第k大的数时入队
				{
				q.pop();
				q.push(a);
				}
				else if(i<k)//队列中少于k个数时入队
				{
					q.push(a);
					i++;
				}
			}
		}
	}

return 0;
	
}