【数据结构作业】堆排

写了个简短的堆排函数，不知道什么时候用的上，先放着。 

#include <bits/stdc++.h>
using namespace std;
#define FOR(i,a,b) for(int i=(a), (i##i)=(b); i<=(i##i); ++i)
#define ROF(i,a,b) for(int i=(a), (i##i)=(b); i>=(i##i); --i)
#define ls o<<1
#define rs o<<1|1
const int N=1e5+5;
int n,a[N];

inline void work(int i,int n){ //下潜函数，用来维护max堆
    for(int o=i*2; o<=n; o=ls){ //不断往下找儿子，如果没有左儿子，说明没有儿子，结束
        if(o+1<=n && a[o+1]>a[o]) o++; //如果有右儿子，看看右儿子是不是更大
        if(a[o]>a[i]){swap(a[i],a[o]); i=o;} //执行交换操作，并且当前节点走到该儿子处
        else break; //如果儿子不是更大，下潜结束
    }
}
inline void Sort(int*a,int n){
    for(int i=n/2; i>=1; i--){ //自下而上下潜，建立max堆
        work(i,n);
    }
    ROF(i,n,2){ //不断把堆顶放到数组后面，并维护max堆
        swap(a[1],a[i]);
        work(1,i-1);
    }
}
signed main(){
    ios::sync_with_stdio(false), cin.tie(0), cout.tie(0);
    cin>>n; FOR(i,1,n) cin>>a[i];
    Sort(a,n);
	FOR(i,1,n) cout<<a[i]<<' '; cout<<endl;
}
// Input: 20 41 67 34 0 69 24 78 58 62 64 5 45 81 27 61 91 95 42 27 36
// Output: 0 5 24 27 27 34 36 41 42 45 58 61 62 64 67 69 78 81 91 95