HDU 4006 The kth great number

关注

原创最新推荐文章于 2026-04-07 09:04:33 发布 1.5k 阅读 AI 写同款· GEO 优化 ›

挑战第七天：Llamaindex RAG 实践安装python 依赖包，并安装 Llamaindex和相关包以及Sentence Transformer 模型。进入开发机后，创建新的conda环境，命名为。镜像，然后在资源配置中，使用。的选项，然后立即创建开发机器。后，点击选择镜像使用。然后安装相关基础依赖。阅读详情

这题的关键在于理解：要求第K大数，那么我们只保留前K个大数，并且按降序排列。这也就是说每加入一个数就找到这个数的位置。

然后将大于K个元素之外的数删除。

利用优先级队列就可以很好的做到这一点。

下面的代码中用到了两种优先级队列的写法。注释的和非注释的都可。并且时间都是62MS。

AC代码：

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

/*
struct node
{
	int tmp;

	friend bool operator < (const node a,const node b)
	{
		return a.tmp>b.tmp;
	}
};
*/

int main()
{
	int n,k,i,num;
	//node num;
    char option;

	while(scanf("%d%d",&n,&k)!=EOF)
	{
		//priority_queue<node> myQueue;
		priority_queue<int,vector<int>,greater<int> > myQueue;

		for(i=1;i<=n;i++)
		{		
			scanf(" %c",&option);
			
			if(option=='I')
			{
				scanf("%d",&num);
				myQueue.push(num);

				if(myQueue.size()>k)
					myQueue.pop();
			}
			else
				printf("%d\n",myQueue.top());
		}
	}

	return 0;
}