#include <cstdio>
int main() {
void HeapSort(int *a, int len);
const int MAXLEN = 10;
int a[MAXLEN] = {3, 8, 9, 1, 0, 2, 5, 7, 4, 6};
printf("befor sort:");
for(int i=0; i<MAXLEN; ++i) {
printf("%d ", a[i]);
}
putchar('\n');
HeapSort(a, MAXLEN);
printf("after sort:");
for(int i=0; i<MAXLEN; ++i) {
printf("%d ", a[i]);
}
putchar('\n');
return 0;
}
void SwapInt(int *e1, int *e2) {
int tmp;
tmp = *e1;
*e1 = *e2;
*e2 = tmp;
}
#define PARENT(i) ((i+1)>>1) - 1
#define LEFT(i) ((i+1)<<1) - 1
#define RIGHT(i) (i+1)<<1
void Heapify(int *a, int index, int len) {
int left = LEFT(index);
int right = RIGHT(index);
int max = index;
if (left<len && a[left]>a[max]) max = left;
if (right<len && a[right]>a[max]) max = right;
if (max != index) {
SwapInt(&a[max], &a[index]);
Heapify(a, max, len);
}
}
void BuildHeap(int *a, int len) {
for (int i=(len-1)>>1; i>=0; --i) {
Heapify(a, i, len);
}
}
void HeapSort(int *a, int len) {
BuildHeap(a, len);
for (int i=len-1; i>0; --i) {
SwapInt(&a[0], &a[i]);
Heapify(a, 0, i);
}
}
要点:
1、使用连续的空间来实现“堆”(堆是完全二叉树),这样的计算 PARENT LEFT RIGHT 会方便
2、 理解Heapify 函数, 三数比较至少需要两次,还有递归的使用
3、理解BuilHeap 使用 (len-1)>>1, 因为可以证明区间( (len-1)>>1, len )所有的元素都是叶子
4、理解什么大顶堆、小顶堆的定义