hdu 4006 The kth great number 优先队列

最新推荐文章于 2020-03-30 22:24:05 发布

原创最新推荐文章于 2020-03-30 22:24:05 发布 · 518 阅读

0 ·

CC 4.0 BY-SA版权

算法专栏收录该内容

123 篇文章

订阅专栏

本文介绍了一种利用优先队列（小顶堆）来高效寻找数组中第K大的元素的方法。通过不断维护一个大小为K的优先队列，确保队列中始终保存数组中最大的K个数。当遍历完所有元素后，队列头部即为第K大的数。文章提供了一个C++示例程序，演示了如何通过输入操作（'I'插入元素，'D'获取当前第K大的数）来动态更新和查询第K大元素。

用优先队列(小的在队头)，只保存最大的k个数。队头是k个数中最小的，就是第k大。

#include<stdio.h>
#include<queue>
using namespace std;

struct node{
	int val;
	bool operator <(node T)const{
		return val>T.val;
	}
}t;
int main(){
	int n,k,val;
	char op[2];
	while(~scanf("%d%d",&n,&k)){
		priority_queue<node> q;//优先队列（队头>其他数）
		for(int i=0;i<n;i++){
			scanf("%s",&op);
			if(op[0]=='I'){
				scanf("%d",&val);
				t.val=val;
				q.push(t);
				if(q.size()>k)
					q.pop();
			}
			else
				printf("%d\n",q.top());//不写等于调用第1个变量(想想指针)
		}
	}
}