#include <iostream>
using namespace std;
void swap(int *a,int *b){ //交换函数
int temp = *a;
*a = *b;
*b = temp;
}
void sheap(int a[],int n,int i){ //维护堆
int current = i;
int rson = i*2+2;
int lson = i*2+1;
//将一个父节点和两个子节点最大的数放到父节点,此时父节点以上都为有序
if(rson < n && a[rson]>a[current]){
current = rson;
}
if(lson < n && a[lson]>a[current]){
current = lson;
}
//若换过位置,则继续将换过的子节点作为父节点向下维护
if(a[current]!=a[i]){
swap(&a[current],&a[i]);
sheap(a,n,current);
}
//整个函数的基础是堆中只有一个位置出了问题,其他地方都是正常的,即父节点大于两个子节点
}
void heap_sort(int a[],int n){
int i;
//建堆
for(i = n/2-1;i>=0;i--){
sheap(a,n,i);
}
//自上向下建堆,保证了堆的正常
//
//排序
//将最大的一位arr[0]放到最后一位,将原数组长度-1,即取出堆,然后自上而下排序,此时也只是
//第一位位置错误,符合sheap()的条件
for(int j= n-1;j>0;j--){
swap(&a[0],&a[j]);
sheap(a,j,0);
}
}
int main(){
int arr[]={3,2,5,6,8,3,4,5};
heap_sort(arr,8);
for(int i = 0;i<8;i++){
cout << arr[i];
}
}
具体的都在代码的注释里,主要是理解递归的思想,还有堆维护,建堆,排序的思路。
具体参考上个文章里那位up的视频。