堆的调整算法(堆排序)及top k问题

算法示例

向上调整

当前节点比父亲大，则与父亲进行交换

void adjustup(HPDataType* x, HPDataType now) {
	int parent = (now - 1) / 2, child = now;
	while (child > 0) {
		if (x[child] > x[parent]) {
			swap(&x[child], &x[parent]);
			child = parent;
			parent = (child - 1) / 2;
		}
		else
			break;
	}
}

向下调整

当前节点比孩子小，则与孩子进行交换

void adjustdown(HPDataType* x,HPDataType parent, HPDataType size) {
	int child=parent*2+1;
	while (child < size) {
		if (x[parent * 2 + 2] > x[child])
			child = parent * 2 + 2;
		if (x[child] > x[parent]) {
			swap(&x[child], &x[parent]);
			parent = child;
			child = parent * 2 + 1;
		}
		else
			break;
	}
}

堆排序

void HeapSort2(int* a, int n) {    //由最后一个父节点往上使用向下堆
	int i;
	for (i =(n-2)/2; i > 1; i--)
		adjustdown(a, i,n);
	for (i = n - 1; i > 0; i--) {
		swap(&a[0], &a[i]);
		adjustdown(a,0, i - 1);
	}
}

调整算法的使用(堆排序)

向上调整要求调整节点为堆的孩子节点

向下调整要求调整节点为堆的父节点

我们通过使用向下调整算法，整理成堆的时间复杂度仅为O(N)

思路

先将原数组转换为堆

从最后一个节点的父节点开始使用向下调整算法直到达到根节点

再

将根节点与最后一个节点进行交换，size-1，使最大（小）的节点放到当前数组的最后位置。

再对根节点进行向下调整重新成堆。

top k 问题

描述：

如何使用很小的空间得到最大（小）的k个数

思路

先导入k个数导入数组，使用向下调整算法成堆

以最大的k个数为例

制造小堆

若之后的数比根节点大，则使用该数覆盖掉根节点，向下调整重新成堆

因为是小堆，所以根节点为最小的数，最小的数一直被覆盖，最后留下的就是最大的k个数

代码

void CreateNDate()
{
	// 造数据
	int n = 10000;
	srand(time(0));
	const char* file = "data.txt";
	FILE* fin = fopen(file, "w");
	if (fin == NULL)
	{
		perror("fopen error");
		return;
	}

	for (size_t i = 0; i < n; ++i)
	{
		int x = rand() % 1000000;
		fprintf(fin, "%d\n", x);
	}

	fclose(fin);
}

void PrintTopK(int k) {
	int* arr = (int*)calloc(k,sizeof(int));
	int a;
	FILE* mn = fopen("data.txt", "r");
	while ((fscanf(mn, "%d", &a)) == 1) {
		if (a > arr[0]) {
			arr[0] = a;
			adjustdown(arr, k);
		}
	}
	for (a = 0; a < k; a++)
		printf("%d ", arr[a]);
}