算法题目——第K大的数

最新推荐文章于 2025-12-25 20:17:35 发布

原创最新推荐文章于 2025-12-25 20:17:35 发布 · 119 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#算法 #数据结构 #c语言

算法题目刷题专栏收录该内容

23 篇文章

订阅专栏

该博客主要介绍了如何利用C++中的优先队列（priority_queue）来解决找到数组中第K大元素的问题。通过构建大顶堆，将数组元素依次插入，然后弹出前K-1个最小元素，保留的最后一个元素即为第K大的数。代码示例展示了具体的实现过程。

题目链接：剑指offer.76

思路：
使用 priority_queue 构建优先队列，这里注意要构建大顶堆，使less；
pop() 弹出前 K-1 个数值
top() 即可

#include<iostream>
#include<vector>
#include<algorithm>
#include<queue>
using namespace std;
/*
    使用 priority_queue 构建优先队列，这里注意要构建大顶堆，使用 less<int>；
    pop() 弹出前 K-1 个数值
    top() 即可
*/
int solve(vector<int> vec, int k)
{
	//小根堆的固定写法
	//priority_queue<int,vector<int>,greater<int>> pq;
	//大根堆的固定写法(默认也为大根堆) 
	priority_queue<int ,vector<int>,less<int>>pq; 
	for(auto a:vec)
	{
		pq.push(a);	
	}
	//先pop出K-1个元素，则第k 个就可以直接输出了 
	 for(int i=0;i<k-1;i++)
	 {
	 	pq.pop();
	 }
	return pq.top();
}
int main()
{
	vector<int> vec;
	int temp;
	int k; 
	while(scanf("%d",&temp)!=EOF)
	{
		vec.push_back(temp);
	}
	cout<<"输入k:"; 
	cin>>k;
	cout<<solve(vec,k); 
 }