目录
1.堆排序
1.1 向上调整建堆
1.2 向下调整建堆
2. 两种建堆方式的时间复杂度比较
2.1 向下调整建堆的时间复杂度
2.2 向上调整建堆的时间复杂度
Topk问题
上节内容,我们讲了堆的实现,同时还包含了向上调整法和向下调整法,最后我们用堆实现了对数据的排序:
int main()
{
HP hp;
HeapInit(&hp);
int arr[] = { 65,100,70,32,50,60 };
int i = 0;
for (i = 0; i < sizeof(arr) / sizeof(int); i++)
{
HeapPush(&hp, arr[i]);
}
while (!HeapEmpty(&hp))
{
HeapDatatype top = HeapTop(&hp);
printf("%d ", top);
HeapPop(&hp);
}
return 0;
}
那以上代码能实现对数据的排序吗?
答案是可以的,但是以上方式有两个弊端:
1. 要先写一个堆,太麻烦
2. 空间复杂度+拷贝数据。
1.堆排序
上节内容中,用堆对数据进行排序,是将数据一个一个插入堆,然后再调整排序的,那我们能不能直接把数据就建成一个堆?
当然可以,建堆有两种方式:向上调整建堆、向下调整建堆。
1.1 向上调整建堆
我们先来讲向上调整建堆:
向上调整建堆其实还是插入堆的逻辑,要求前面的数据必须是一个堆,下标从1开始是因为一个数据本身就可以被看做一个堆,然后向上调整。
下图就是我们对一个数组数据进行向上调整建堆后的结果,可以看出来,此时我们建的是一个小堆:
现在问题来了,我们要把数据排为升序,建大堆还是建小堆好?
先说结论:升序 -- 建大堆 降序 -- 建小堆。
假设我们要得到升序,此时又建的是小堆,那我们就把选出的最小的数据放在下标为0的位置,要想继续选出次小的数据放在下标为1的位置,就要把剩下的数据看做堆,这样堆的关系就全乱了,只能重新建堆,代价太大。
而如果我们建大堆,向下调整选出最大的数据,首尾交换,把最大的数据放在最后一个下标的位置,然后隔离最后一个数据,把其他数据看做一个堆,再向下调整选出次大的,首尾交换......直到所有的数据被排好序,此时得到的就是数据升序。
代码如下:
#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
typedef int HeapDatatype;
swap(HeapDatatype* p1, HeapDatatype* p2)
{
HeapDatatype tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
//向上调整法
void AdjustUp(HeapDatatype* a, int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
if (a[parent] < a[child])
{
HeapDatatype p = a[parent];
a[parent] = a[child];
a[child] = p;
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
//向下调整法
void AdjustDown(HeapDatatype* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && a[child] < a[child + 1])
{
child++;
}
if (a[parent] < a[child])
{
swap(&a[parent], &a[child]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
//堆排序
void HeapSort(int* a, int n)
{
//建堆 - 向上调整建堆
for (int i = 1; i < n; i++)
{
AdjustUp(a, i);
}
//向下调整得到次大数据
int end = n - 1;
while (end > 0)
{
swap(&a[0], &a[end]);
AdjustDown(a, end, 0);
end--;
}
}
int main()
{
int a[] = { 7,8,3,5,1,9,5,4 };
HeapSort(a, sizeof(a) / sizeof(int));
return 0;
}
我们建的是大堆,最后得到的就是升序:
要得到数据降序,就要建小堆,向下调整选出最小的数据,首尾交换,把最小数据放在最后一个下标的位置,隔离最后一个数据,把其他数据看做一个堆,再向下调整选出次小的数据,首尾交换......直到所有数据都被拍好序,这就得到数据降序。
代码如下:(只需将向下调整和向上调整中的'<'改为'>'即可)
#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
typedef int HeapDatatype;
swap(HeapDatatype* p1, HeapDatatype* p2)
{
HeapDatatype tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
//向上调整法
void AdjustUp(HeapDatatype* a, int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
if (a[parent] > a[child])
{
HeapDatatype p = a[parent];
a[parent] = a[child];
a[child] = p;
child = parent;
parent = (child - 1) / 2;
}
else
{
break;
}
}
}
//向下调整法
void AdjustDown(HeapDatatype* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && a[child] > a[child + 1])
{
child++;
}
if (a[parent] > a[child])
{
swap(&a[parent], &a[child]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
//堆排序
void HeapSort(int* a, int n)
{
//建堆 - 向上调整建堆
for (int i = 1; i < n; i++)
{
AdjustUp(a, i);
}
//向下调整得到次小数据
int end = n - 1;
while (end > 0)
{
swap(&a[0], &a[end]);
AdjustDown(a, end, 0);
end--;
}
}
int main()
{
int a[] = { 7,8,3,5,1,9,5,4 };
HeapSort(a, sizeof(a) / sizeof(int));
return 0;
}
由于我们建的是小堆,所以得到的就是数据降序:
注意:不论是升序还是降序,数据都是从后往前放的,这样就不会使堆的关系混乱。
1.2 向下调整建堆
我们可以看到,堆排序使用向上调整建堆,还要写两个函数:向下调整函数、向上调整函数。
那我们想用一个向下调整函数就解决问题呢?
这就需要向下调整建堆:
向下调整建堆要求根节点的左右子树都是大堆(小堆),如果左右子树不满足大堆,我们只需要确保左右子树的左右子树是大堆(小堆)即可,如果又不是,我们再往下找,所以只要使所有父节点的左右子树都是大堆(小堆)就行,那我们就倒着调整,因为叶子节点本身就是一个堆,所以不需要调整,那就从最后一个节点的父节点开始调整。
代码如下:
#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
typedef int HeapDatatype;
swap(HeapDatatype* p1, HeapDatatype* p2)
{
HeapDatatype tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
//向下调整法
void AdjustDown(HeapDatatype* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && a[child] > a[child + 1])
{
child++;
}
if (a[parent] > a[child])
{
swap(&a[parent], &a[child]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
//堆排序
void HeapSort(int* a, int n)
{
//建堆 - 向下调整建堆
for (int i = (n - 1 - 1) / 2; i >= 0; i--)
{
AdjustDown(a, n, i);
}
int end = n - 1;
while (end > 0)
{
swap(&a[0], &a[end]);
AdjustDown(a, end, 0);
end--;
}
}
int main()
{
int a[] = { 7,8,3,5,1,9,5,4 };
HeapSort(a, sizeof(a) / sizeof(int));
return 0;
}
代码中int i=(n-1-1)/2是通过孩子找父亲的下标,n是数组大小,先减一得到最后一个下标,再减一除以二得到最后一个孩子的父节点。
这就是向下调整建堆,以后我们用的都是向下调整建堆,不再使用向上调整建堆,这两种方式不仅代码量上有差距,时间复杂度上也有差距,向下调整建堆的时间复杂度更小。
2. 两种建堆方式的时间复杂度比较
2.1 向下调整建堆的时间复杂度
前文中我们知道了,向下调整建堆要保证每个父节点的左右子树都是大堆(小堆),所以我们在调整的时候是从下往上进行的,而最后一层的每个叶节点本身就可以看做一个堆,不用调整,从它们的父节点开始调整(即倒数第二层开始调整),所以时间复杂度如下:
总步数 = ∑(每一层的节点数*该节点需要调整的层数)
2.2 向上调整建堆的时间复杂度
向上调整和向下调整刚好相反,向下调整时,第h-1行的2^(h-2)个节点需向下调整1层,而向上调整时,第h-1行的2^(h-1)个节点需要向上调整h-2,向下调整是大乘小、小乘大,而向上调整时大乘大、小乘小,时间复杂度如下:
以上就是向上调整建堆和向下调整建堆的时间复杂度,那我们整个堆排序的过程的时间复杂度是多少呢?
堆排序过程中,除了建堆还有向下调整选数,当选数时,要首尾交换,交换一次,从头向下调整一次, 所以第h行的2^(h-1)个节点,每次首尾交换时都要调整(h-1)次,一共2^(h-1)*(h-1),由此可见,选数据过程中的时间复杂度和向上调整建堆的时间复杂度保持一致,即为O(N*logN)
所以堆排序整体的时间复杂度是:建堆+选数 = O(N+N*logN),即O(N*logN)。
Topk问题
TOP-K问题:即求数据结合中前K个最大的元素或者最小的元素,一般情况下数据量都比较大。
比如:专业前10名、世界500强、富豪榜、游戏中前100的活跃玩家等。
对于Top-K问题,能想到的最简单直接的方式就是排序,但是:如果数据量非常大,排序就不太可取了(可能数据都不能一下子全部加载到内存中)。最佳的方式就是用堆来解决,基本思路如下:
1. 用数据集合中前K个元素来建堆
前k个最大的元素,则建小堆
前k个最小的元素,则建大堆
2. 用剩余的N-K个元素依次与堆顶元素来比较,不满足则替换堆顶元素
将剩余N-K个元素依次与堆顶元素比完之后,堆中剩余的K个元素就是所求的前K个最小或者最大的元素。
比如:我们要找10000个数中的前K个最小的数,就把先把前K个数建小堆,然后把用剩余的N-K个元素依次与堆顶元素来比较,不满足则替换堆顶元素
代码如下:
#define _CRT_SECURE_NO_WARNINGS 1
#include<stdio.h>
#include<time.h>
#include<stdlib.h>
//Top-K问题
typedef int HeapDatatype;
swap(HeapDatatype* p1, HeapDatatype* p2)
{
HeapDatatype tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
//向下调整法
void AdjustDown(HeapDatatype* a, int n, int parent)
{
int child = parent * 2 + 1;
while (child < n)
{
if (child + 1 < n && a[child] > a[child + 1])
{
child++;
}
if (a[parent] > a[child])
{
swap(&a[parent], &a[child]);
parent = child;
child = parent * 2 + 1;
}
else
{
break;
}
}
}
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)
{
const char* file = "data.txt";
FILE* fout = fopen(file, "r");
if (fout == NULL)
{
perror("fopen error");
return;
}
int* kminheap = (int*)malloc(sizeof(int) * k);
if (kminheap == NULL)
{
perror("malloc error");
return;
}
for (int i = 0; i < k; i++)
{
fscanf(fout, "%d", &kminheap[i]);
}
// 建小堆
for (int i = (k - 1 - 1) / 2; i >= 0; i--)
{
AdjustDown(kminheap, k, i);
}
int val = 0;
while (!feof(fout))
{
fscanf(fout, "%d", &val);
if (val > kminheap[0])
{
kminheap[0] = val;
AdjustDown(kminheap, k, 0);
}
}
for (int i = 0; i < k; i++)
{
printf("%d ", kminheap[i]);
}
printf("\n");
}
int main()
{
CreateNDate();
PrintTopK(5);
return 0;
}
关于堆排序的所有内容已经学完了,下节我们继续讲二叉树的前序、中序、后序和层序。
未完待续。。。