堆排序实现代码

就一枚小白

已于 2022-08-07 15:59:06 修改

阅读量350

点赞数 3

分类专栏：算法文章标签：算法 c++ 数据结构

于 2022-08-07 15:56:39 首次发布

本文链接：https://blog.csdn.net/qq_52855744/article/details/126212395

版权

算法专栏收录该内容

16 篇文章 0 订阅

订阅专栏

降序

构建最小堆

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int arr[N];

void adjust(int a[], int n, int i){
    int mi = i, l = i * 2 + 1, r = i * 2 + 2;
    if(l < n && a[mi] > a[l]) mi = l;
    if(r < n && a[mi] > a[r]) mi = r;
    if(mi != i){
        swap(a[mi], a[i]);
        adjust(a, n, mi);
    }
}

void buildMinHeap(int a[], int n){
    for(int i = n / 2 - 1; i >= 0; i--){
        adjust(a, n, i);
    }
    for(int i = n - 1; i >= 0; i--){
        swap(a[0], a[i]);
        adjust(a, i, 0);
    }
}

int main(){
    int n;
    cin >> n;
    for(int i = 0; i < n; i++) cin >> arr[i];
    buildMinHeap(arr, n);
    return 0;
}

升序

构建最大堆

#include <iostream>
using namespace std;

const int N = 1e5 + 10;
int arr[N];

void adjust(int a[], int n, int i){
    int ma = i, l = i * 2 + 1, r = i * 2 + 2;
    if(l < n && a[ma] < a[l]) ma = l;
    if(r < n && a[ma] < a[r]) ma = r;
    if(ma != i){
        swap(a[ma], a[i]);
        adjust(a, n, ma);
    }
}

void buildMinHeap(int a[], int n){
    for(int i = n / 2 - 1; i >= 0; i--){
        adjust(a, n, i);
    }
    for(int i = n - 1; i >= 0; i--){
        swap(a[0], a[i]);
        adjust(a, i, 0);
    }
}

int main(){
    int n;
    cin >> n;
    for(int i = 0; i < n; i++) cin >> arr[i];
    buildMinHeap(arr, n);
    return 0;
}