STL栈、队列、堆的算法笔记（4）——【数组中第K大的数】

最新推荐文章于 2021-09-06 14:16:21 发布

小小雨伞

最新推荐文章于 2021-09-06 14:16:21 发布

阅读量211

点赞数

CC 4.0 BY-SA版权

分类专栏：算法文章标签： stl 堆

本文链接：https://blog.youkuaiyun.com/qq_42794826/article/details/99670174

算法专栏收录该内容

62 篇文章

订阅专栏

本文介绍了一种高效算法，用于查找未排序数组中的第K大数字。通过使用最小堆，我们可以在O(n log k)的时间复杂度内解决问题。文章包含了一个C++实现示例，演示了如何利用优先队列来维护前K个最大元素。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

数组中第K大的数

题目:

已知一个未排序的数组，求这个数组中第K大的数字。
如，array = [3,2,1,5,6,4] ， k = 2，return 5

思路图解:

在这里插入图片描述

解题代码:

#include <iostream>
#include<queue>
using namespace std;
class MyClass
{
public:
	int findKthNum(vector<int>& nums, int k) {
		//维护最小堆
		priority_queue<int, vector<int>, greater<int>> q;
		for (int i = 0; i < nums.size(); i++)
		{
			if (q.size()<k)
			{
				q.push(nums[i]);
			}
			else if (q.top() < nums[i])
			{
				q.pop();
				q.push(nums[i]);
			}
		}
		return q.top();
	}
};
int main()
{
	vector<int> nums;
	nums.push_back(3);
	nums.push_back(2);
	nums.push_back(1);
	nums.push_back(5);
	nums.push_back(6);
	nums.push_back(4);
	MyClass cls;
	int x = cls.findKthNum(nums, 2);
	cout << "第二大的数为: " << x << endl;
}