堆排序,时间复杂度O(nlogn),空间复杂度O(1),就地算法。
#include <bits/stdc++.h>
using namespace std;
void HeapFin(int *a,int k,int n){ //堆维护算法
while(k < n){
int left = 2*k + 1 , right = 2*k + 2;
int Max = a[k] , idx = k;
if(left < n && a[left] > Max){
Max = a[left] ;
idx = left ;
}
if(right < n && a[right] > Max){
Max = a[right] ;
idx = right ;
}
if(Max != a[k]){ //孩子中有值比当前节点大的
swap(a[k],a[idx]) ;
k = idx;
}else break;
}
}
void CreateHeap(int *a,int n){
for(int i = n/2;i>=0;--i)
HeapFin(a,i,n);
}
void HeapSort(int *a,int n){
CreateHeap(a,n); //创建最大堆
int count = n-1;
for(int i=1;i<n;++i){ //n-1遍,排序n-1个数结束
swap(a[0],a[count--]) ; //每次选择一个数后,需要让数组长度-1
HeapFin(a,0,count+1) ; //维护最大堆的性质
}
}
int main(){
int a[20] ;
for(int i=0;i<20;++i){
a[i] = rand()% 100 ;
cout<<a[i] <<' ';
}
cout<<endl;
HeapSort(a,20) ;
for(int i=0;i<20;++i)
printf("%d%c",a[i],i==19?'\n':' ' );
return 0;
}