堆排序
HeapSort(a, n)的作用是:对数组a进行升序排序;其中,a是数组,n是数组长度。
HeapSort(a, n)的操作分为两部分:初始化堆 和 交换数据。
MaxHeapDown(a, start, end)是最大堆的向下调整算法。
#include<iostream>
using namespace std;
void MaxHeapDown(int arr[], int start, int end){
int parent = start;
int child = parent * 2 + 1;//记录当前节点的子节点
for(; child <= end; child = parent*2 + 1) {//向下调整
if(child < end && arr[child+1] > arr[child]) {//选择左右孩子中最大的一个
child++;
}
if(arr[child] > arr[parent]) {//如果孩子节点大于父节点交换位置继续向下调整
swap(arr[parent], arr[child]);
parent = child;
}else {//否则所有子树中的节点都比父节点小,停止向下调整
break;
}
}
}
void HeapSort(int arr[], int n) {
//初始化堆 构造大顶堆
for(int i = n / 2 - 1; i >= 0; i--){
MaxHeapDown(arr, i, n-1);
}
// printf("\n%d\n", arr[0]);
for(int j = n-1; j > 0; j--){
swap(arr[0], arr[j]); //将大顶堆的堆顶元素与最后一个元素交换位置
// cout << arr[0] << " " << arr[j] << endl;
//重新构造大顶堆
MaxHeapDown(arr, 0, j-1);
}
}
int main () {
int a[] = {20,30,90,40,70,110,60,10,100,50,80};
int len = sizeof(a)/sizeof(a[0]);
printf("Before: ");
for(int i = 0; i < len; i++){
printf("%d ", a[i]);
}
HeapSort(a, len);
printf("\nAfter: ");
for(int i = 0; i < len; i++){
printf("%d ", a[i]);
}
return 0;
}