C++算法入门练习——数据流第K大元素

现有一个初始为空的序列S,对其执行n个操作,每个操作是以下两种操作之一:

  1. 往序列S中加入一个正整数x;
  2. 输出当前序列S​中第k​大的数。

其中,第k大是指将序列从大到小排序后的第k个数。

C++算法入门练习——数据流第K大元素_第1张图片

利用stl里的priority_queue自动实现最大堆排序。

这里用个小技巧,输入时,如果已经当前堆有了k个数,那么再输入一个数后,要弹出,这样就可以保持堆里只包含k个数,而堆顶的数就是第k大的数。

完整代码如下:

#include 
#include 
#include 
#include 
#include 
using namespace std;

int main(){
	priority_queue,greater> q;
	int n,k;
	cin>>n>>k;
	string stemp;
	int temp; 
	for(int i=0;i>stemp;
		if(stemp=="Push"){
			cin>>temp;
			q.push(temp);
			if(q.size()>k){
				q.pop();
			}
		}
		else{
			if(q.size()>=k){
				temp = q.top();
				cout<

你可能感兴趣的:(c++,开发语言,算法)