目录
问题:已知我们有一个存放数据的数组,我们要使它的数据由小到大升序排序,要求运用堆排序算法实现
堆排序,顾名思义就是我们要将数组里的数据建堆,但究竟建大堆还是小堆呢?
问题:已知我们有一个存放数据的数组,我们要使它的数据由小到大升序排序,要求运用堆排序算法实现
堆排序,顾名思义就是我们要将数组里的数据建堆,但究竟建大堆还是小堆呢?
思路:因为是升序排序,所以我们建大堆,使最大的数据在第一个根节点的位置,建好之后,不断地将最大的数据移动到数组的最后一个元素的位置,使最大值交换到数组的最后面,然后将总体有效数据个数-1,使得再次访问堆,访问不到已经移动好的数据。
建堆算法:
方法一:从前往后向上调整建堆。
方法二:从后往前向下调整建堆。(运用向下调整算法的前提是:在此节点的下方是大堆或是小堆,而我们要实现大堆,所以要求是此节点的子树为大堆才可以)
问题:这两个方法哪个更好呢?
答:方法二好,因为如果按方法一的算法的话,这个二叉树的最后一层向上调整的层数是最多的;而如果按方法二的话,最后一层都不用去向下调整,这大大增加了效率。所以我们讲解堆排序用方法二来实现。
堆排序中要用到的代码:
typedef int HPDataType;
typedef struct Heap
{
HPDataType* a;
int size;
int capacity;
}HP;
//方便交换的函数
void Swap(HPDataType* p1, HPDataType* p2)
{
HPDataType tmp = *p1;
*p1 = *p2;
*p2 = tmp;
}
//向上调整
void AdjustUp(HPDataType* a, int child)
{
int parent = (child - 1) / 2;
while (child > 0)
{
if (a[parent] < a[child])
{
Swap(&a[parent], &a[child]);
child = parent;
parent = (child - 1) / 2;
}
else
break;
}
}
//向下调整
void AdjustDown(HPDataType* a, int n, int parent)
{
int child = 2 * parent + 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 = 2 * parent + 1;
}
}
堆排序的代码实现:
//堆排序
void HeapSort(HPDataType* a, int n)
{
//建堆算法
int tail = 0;
for (tail = (n-1-1)/2;tail >= 0;tail--)
{
AdjustDown(a, n, tail);
}
//建好大堆了
//堆排序
int size = n;
while (size > 1)
{
Swap(&a[0], &a[size - 1]);
size--;
AdjustDown(a, size, 0);
}
}