C++模板堆排序

C++用模板函数实现堆排序

代码

#include<iostream>
#include<set>
#include<map>
#include<algorithm>
using namespace std;
template<typename T>
void adjustHeap(T *k,T i,T n) {  //i为结点位置 n为数组大小 
    int temp=k[i]; //这是此结点的值  堆排序比较顺序 从左往右 从上到下
    int j=i*2;     //左结点  注意数组从0开始 左结点 2*i+1  数组从1开始那么就是2*i
    while(j<=n) {
        if(j<n&&k[j]<k[j+1]) {
            //如果左结点比右结点小  那么移动到右结点
            j++; 
        }
        if(temp>k[j]) {
            //如果比父结点小，那么退出
             break;
        }
        k[j/2]=k[j];  //大的值赋值给结点
        j=j*2; 
    }
    k[j/2]=temp;

}
int main() {
    //大堆排序的目的就是将最大的值放在树顶部，然后和最后一个元素交换位置，然后继续找出最大值和n-1交换位置
    //这样就能形成一个有序的数组
    int N;
    cin>>N;
    int a[N];
    for(int i=1;i<=N;i++) {
        cin>>a[i];
    }
    //创建堆 
    for(int i=N/2;i>0;i--) {
        adjustHeap(a,i,N);
    }
    int temp;
    for(int i=N;i>0;i--) {
        temp=a[i];
        a[i]=a[1];
        a[1]=temp;
        adjustHeap(a,1,i-1);
    }
    cout<<' ';
    for(int i=1;i<=N;i++) {
        cout<<a[i]<<' ';
    }
    return 0;
 }