堆TOP K问题

问题描述: 以找最大的k个数为例

如果想要找出一组数据中的最大的k个值,如果直接建堆时间复杂度时O(n*logn),当数据量大的时候,不仅时间复杂度高,而且空间复杂度也很高为O(n),加入数据是1亿个,就会占用大量的空间。

解决方法:

建立一个k个数的小根堆,因为是找最大的k个数,所以要建立小根堆,这样每次都是当前最大的k个数里面最小的在堆顶,当前的数和堆顶比较,如果比堆顶大,就和堆顶交换,然后向上调整,如此循环最后得到最大的k个数。时间复杂度是O(n*logK),空间复杂度是O(1)。注意不是建立大根堆,如果建立大根堆,那么堆顶如果是最大的,那么就会一直卡着,后面的数据无法进入到堆里面。

如果要是求最小的前k个就是建立大堆

#include "Heap.h"
void CreatData()
{
	int n = 10000;
	srand(time(0));
	const char* file = "data.txt";
	FILE* fin = fopen(file, "w");
	if (fin == NULL)
	{
		perror("fopen file");
		exit(-1);
	}
	for (int i = 0; i < n; ++i)
	{
		int x = (rand() + i) % 10000000;
		fprintf(fin, "%d\n", x);
	}
	fclose(fin);
}
void PrintTopK(const char* file, int k)
{
	FILE* fout = fopen(file, "r");
	if (fout == NULL)
	{
		perror("fopen file");
		exit(-1);
	}
	int* minheap = (int*)malloc(sizeof(int) * k);//开辟一块空间存堆
	if (fout == NULL)
	{
		perror("malloc file");
		exit(-1);
	}
	for (int i = 0; i < k; ++i)
	{
		fscanf(fout, "%d", &minheap[i]);
		AdjustUP(minheap, i);
	}
	int x = 0;
	while(fscanf(fout, "%d", &x) != EOF)
	{
		if (x > minheap[0])
		{
			minheap[0] = x;
			AdjustDown(minheap, k, 0);
		}
	}
	for (int num = 0; num < k; ++num)
	{
		printf("%d ", minheap[num]);
	}
}
int main()
{
	//CreatData();
	PrintTopK("data.txt", 5);
	return 0;
}
  • 7
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值