我一直在努力学习如何实施heapsort。
特别是,我有一个堆栈算法,按照升级顺序对输入进行排序, 但我无法弄清楚需要改变什么以使其降序。
以下是代码:(部分内容来自教科书)
void heapSort(int numbers[], int array_size)
{
int i, temp;
for (i = (array_size / 2); i >= 0; i--)
siftDown(numbers, i, array_size - 1);
for (i = array_size-1; i >= 1; i--)
{
temp = numbers[0];
numbers[0] = numbers[i];
numbers[i] = temp;
siftDown(numbers, 0, i-1);
}
}
void siftDown(int numbers[], int root, int bottom)
{
int done, maxChild, temp;
done = 0;
while ((root*2 <= bottom) && (!done))
{
if (root*2 == bottom)
maxChild = root * 2;
else if (numbers[root * 2] > numbers[root * 2 + 1])
maxChild = root * 2;
else
maxChild = root * 2 + 1;
if (numbers[root] < numbers[maxChild])
{
temp = numbers[root];
numbers[root] = numbers[maxChild];
numbers[maxChild] = temp;
root = maxChild;
}
else
done = 1;
}
}
如果有人能够指出需要更改代码的哪一部分,那将非常有用:)。
答案 0 :(得分:2)
您实现的是“最大堆”(即每个节点的子节点的值小于父节点的值)。您只需更改siftDown
函数,以便父级始终小于子级。
这使得它成为一个小堆并以相反的顺序提供值(因为最小的元素首先到达堆的顶部),因此给你一个降序排序。