需要升序排列元素,建立大根堆;需要降序排序元素,建立小根堆。
(1)堆调整的过程:如果子树的根小(大)于左右孩子的较大者,那么将较大(小)者与根进行交换,交换后较大(小)者位置为根的子树是否构成子树堆;如果构成子树堆,则子树调整完成;否则,重复上述步骤,继续对不满足堆的子树进行调整。
(2)建立大(小)根堆:从右到左,从下到上来访问每棵子树的根,然后将子树传入调整函数,将其调整为子树堆。
(3)堆排序:大(小)根堆堆顶元素为最大(小)值,输出堆顶元素,将堆底元素送人堆顶,然后将剩余元素整理成堆;重复上述步骤,直到将全部元素有序输出。
#define _CRT_SECURE_NO_WARNINGS
#include<stdio.h>
#include<stdlib.h>
#include<string.h>
typedef int ElemType;
#define MaxSize 100
/*需要升序排列元素,建立大根堆;需要降序排序元素,建立小根堆*/
void Swap(int& a, int& b) {
int temp = a;
a = b;
b = temp;
}
/*堆调整的过程:如果子树的根小于左右孩子的较大者,那么将较大者与根进行交换,交换后较大者位置为根的子树是否构成子树堆;
如果构成子树堆,则子树调整完成;否则,重复上述步骤,继续对不满足堆的子树进行调整。*/
void AdjustHeap(int A[], int length, int index) {
int posIndex = index;
int Elem = A[index];
while (2 * posIndex <=length) {
int lIndex = 2 * posIndex;
int rIndex = 2 * posIndex + 1;
int bigger = lIndex;//较大者初始化为左孩子
if (rIndex <= length &&A[rIndex] > A[bigger]) {//存在右孩子且右孩子为较大则
bigger = rIndex;//更新较大者为右孩子
}
//不满足堆的定义,需要调整
if (Elem < A[bigger]) {//注意这里是用Elem元素进行比较
A[posIndex] = A[bigger];//将较大者放入根结点中
posIndex = bigger;//更新根节点位置,继续进行调整
}
else {//满足堆的定义,不需要调整,结束循环
break;
}
}
A[posIndex] = Elem;//将根元素写入最终位置
}
/*建立大根堆:从右到左,从下到上来访问每棵子树的根,然后将子树传入调整函数,将其调整为子树堆*/
void CreatHeap(int A[],int length) {
for (int index =length / 2; index > 0; index--) {
AdjustHeap(A,length, index);
}
}
/*堆排序:大根堆堆顶元素为最大值,输出堆顶元素,将堆底元素送人堆顶,然后将剩余元素整理成堆;重复上述步骤,直到将全部元素有序输出*/
void HeapSort(int A[],int length) {
CreatHeap(A,length);//建立堆
for (int i = length; i > 1; i--) {//共需要n-1趟交换和建堆
Swap(A[1],A[i]);//输出堆顶元素(和堆底元素进行交换)
AdjustHeap(A, i - 1, 1);//将剩余i-1个元素调整成堆
}
}
int main() {
int len, A[MaxSize];
scanf("%d", &len);
for (int i = 1; i <= len; i++) {
scanf("%d", &A[i]);
}
HeapSort(A, len);
for (int i = 1; i <= len; i++) {
printf("%d ", A[i]);
}
return 0;
}