主要包括两个过程:1.建堆 2.向下调整
建堆过程: 从下往上进行筛选(类似于打比赛),将最大值或者最小值调整到 顶部。
调整:将顶部的最大值或最小值 放到数组的最后,下次建堆就不用这个元素了。
重复以上过程直到筛选的区间为1
堆的表示:数组的形式,第i个元素的左子节点为2*i 右节点为 2*i+1,数组的先后顺序为层序遍历的顺序。
代码:
#include <iostream>
#include <vector>
using namespace std;
void build_heap(vector<int> &r,int n);
void heap_sort(vector<int> &r){
for(int n=r.size();n>1;n--){
build_heap(r,n);
r[0] ^= r[n-1];
r[n-1] ^= r[0];
r[0] ^= r[n-1];
}
}
void build_heap(vector<int> &r,int n){
for(int i=n/2;i>=0;i--){
if(2*i<n &&r[i]<r[2*i] && r[2*i]>r[2*i+1]){
r[i] ^= r[2*i];
r[2*i] ^= r[i];
r[i] ^= r[2*i];
}
if(2*i+1<n &&r[i]<r[2*i+1] && r[2*i]<r[2*i+1]){
r[i] ^= r[2*i+1];
r[2*i+1] ^= r[i];
r[i] ^= r[2*i+1];
}
}
}
void print(vector<int> &r){
for(int i=0;i<r.size();i++){
std::cout<<r[i]<<" ";
}
std::cout<<std::endl;
}
int main(){
vector<int> r = {7,4,3,1,6,8,10,-3};
heap_sort(r);
print(r);
return 0;
}