【C语言】【堆排序实现TOPK问题】写一个堆排序,并且在一百万个数中找出最大的前K个

1.堆排序的实现:

如果要升序输出,则实现大堆
如果要降序输出,则实现小堆

逻辑:(升序输出)
将堆顶的元素和最后一个元素交换位置,此时左子树和右子树分别仍是大堆的顺序,交换位置后选出了最小的数。接着向下调整使剩余的元素再成为一个大堆,然后再交换,选出第二小的数…
【C语言】【堆排序实现TOPK问题】写一个堆排序,并且在一百万个数中找出最大的前K个_第1张图片
【C语言】【堆排序实现TOPK问题】写一个堆排序,并且在一百万个数中找出最大的前K个_第2张图片
【C语言】【堆排序实现TOPK问题】写一个堆排序,并且在一百万个数中找出最大的前K个_第3张图片

在这里插入图片描述

代码:
Heap.h:

#include
#include
#include
#include
void Swap(int* p1, int* p2);
void AdjustDown(int* a, int n,int parent);
void AdjustUp(int* a, int child);
void HPSort(int* a, int n);

Heap.c:

#include
#include
#include
#include"Heap.h"

void Swap(int* p1, int* p2)
{
	int tmp = *p1;
	*p1 = *p2;
	*p2 = tmp;
}
//向下调整:(建小堆)
void AdjustDown(int* a, int n,int parent)
{
	int child = 2 * parent + 1;
	while (child < n)
	{
		if (child + 1 < n && a[child + 1] > a[child])
		{
			child = child + 1;
		}
		if (a[parent] < a[child])
		{
			Swap(&a[parent], &a[child]);
			parent = child;
			child = 2 * child + 1;
		}
		else
			break;
	}
}
//向上调整:(建大堆)
void AdjustUp(int*a,int child)
{
	assert(a);	
	int parent = (child - 1) / 2;
	while (child != parent)
	{
		if (a[child] > a[parent])
		{
			Swap(&a[child], &a[parent]);
			child = parent;
			parent = (parent - 1) / 2;
		}
		else
			break;

	}

}

//堆排序(升序):
void HPSort(int* a,int n)
{
	int end = n - 1;
	//建立大堆:
	for (int i = 1;i < n;i++)
	{
		AdjustUp(a, i);
	}
	while (end > 0)
	{
		Swap(&a[0], &a[end]);
		AdjustDown(a, end, 0);
		end--;
	}
}

2.TopK问题

将生成的一百万个随机数写入文件里面:
(Heap.h和Heap.c未改动)

#include"Heap.h"
//造数据:
void CreateData()
{
	int n = 1000000;
	srand(time(0));
	const char* file = "data.txt";
	FILE* fin = fopen(file, "w");
	if (fin == NULL)
	{
		perror("fopen error");
		return;
	}
	for (int i = 0;i < n;i++)
	{
		int x = (rand() + i) % 10000000;
		fprintf(fin, "%d\n", x);
	}
	fclose(fin);
}
//打印最大的k个数:
void PrintTopk(const char* file, int k)
{
	FILE* fout = fopen(file, "r");
	if (fout == NULL)
	{
		perror("fopen error");
		return;
	}
	int* minheap = (int*)malloc(sizeof(int) * k);
	int x = 0;
	for (int i = 0;i < k;i++)
	{
		fscanf(fout, "%d", &minheap[i]);
		AdjustUp(minheap, i);
	}
	while (fscanf(fout, "%d", &x) != EOF)
	{
		if (x > minheap[0])
		{
			minheap[0] = x;
			AdjustDown(minheap, k, 0);
		}
	}
	for (int i = 0;i < k;i++)
	{
		printf("%d ", minheap[i]);
	}
	printf("\n");
	free(minheap);
	fclose(fout);
}
int main()
{
	CreateData();
	PrintTopk("data.txt", 5);
	return 0;
}

【C语言】【堆排序实现TOPK问题】写一个堆排序,并且在一百万个数中找出最大的前K个_第4张图片

【C语言】【堆排序实现TOPK问题】写一个堆排序,并且在一百万个数中找出最大的前K个_第5张图片

你可能感兴趣的:(c语言,算法,开发语言,笔记,数据结构)