堆排序主要分为两步:
第一步,将一组无序序列建立起小顶堆或大顶堆(完全二叉树)。
第二步,输出堆顶的元素,然后对剩余的元素重新进行建立堆的过程(也就是第一步)。
由于此代码数组的0位置是用作暂存其他数,所以在输入序列时,0号位不参与排序。
#include<bits/stdc++.h>
using namespace std;
void AdjustDown(int A[],int j,int len){
A[0] = A[j];
for(int i=2*j;i <= len;i *= 2){
if(A[i] < A[i+1] && i < len){ //寻找子节点中较大的数
i++;
}
if(A[i] > A[0]){
A[j] = A[i]; //将A[i]放到双亲结点上
j = i; //将 i 赋值给 j 开始对原来i结点的子节点进行筛选
}
else{
break; //A[0] > A[i] 不必继续
}
}
A[j] = A[0]; //将筛选出的值放入最终位置
}
void BuildMaxHeap(int A[],int len){
for(int i = len/2;i > 0; i--){ //从i = [n/2]~1反复调整堆
AdjustDown(A,i,len);
}
}
void HeapSort(int A[],int len){
BuildMaxHeap(A,len); //初始建堆
for(int i=len;i>1;i--){
swap(A[i],A[1]);
AdjustDown(A,1,i-1); //整理,把剩余的i-1个元素整理成堆
}
}
int main(){
int a[6] = {0,5,3,8,7,9};
HeapSort(a,6);
for(int i=1;i<6;i++){
cout << a[i];
}
}