TOPK问题----代码实现打印数据中前K大的数

最新推荐文章于 2024-10-07 17:14:06 发布

晚风_清扬

最新推荐文章于 2024-10-07 17:14:06 发布

阅读量480

点赞数

CC 4.0 BY-SA版权

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/RUN32875094/article/details/80025358

算法专栏收录该内容

33 篇文章

订阅专栏

本文介绍了一种高效的方法来统计从10万随机整数中选取的最大10个数值。通过使用小根堆数据结构实现了一个简易的Top K算法，此算法能够有效地筛选出指定数量的最大值。

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

/*
问题：
	随机产生100000个整数，范围是【0，1000】，快速统计出值最大的前10个？
*/
#include<iostream>
#include<vector>
#include<queue>
#include<ctime>
#include<functional>
using namespace std;
class TopK{
public:
	TopK()
	{
		srand(time(NULL));//初始化时间种子
		for (int i = 0; i < 100000; ++i)
		{
			v.push_back(rand() % 1000);
		}
	}
	void PrintTopK(int k)
	{
		priority_queue<int, vector<int>, greater<int>> pri;//小根堆数据结构
		
		//先维持一个k个元素的小根堆
		for (int i = 0; i < k; ++i)
		{
			pri.push(v[i]);
		}

		//一次遍历往后遍历每个数
		for (int i = k; i < v.size(); ++i)
		{
			if (v[i] > pri.top())//如果比堆顶的大，就将堆顶的元素出堆，将该元素入堆
			{//因为堆顶的元素是k个元素中最小的，比堆顶大说明可能TOPK个元素之一
				pri.pop();
				pri.push(v[i]);
			}
		}
		//遍历一次后，小根堆中的元素都是所有元素中最大的 K个数
		for (int i = 0; i < k; ++i)
		{
			cout << pri.top() << "  ";
			pri.pop();
		}
	}
	vector<int> v;
};
int main()
{
	TopK t;
	t.PrintTopK(10);
	cin.get();
	return 0;
}