《算法导论》伪代码C++实现--堆排序

最新推荐文章于 2021-02-28 12:21:26 发布

北城垚

最新推荐文章于 2021-02-28 12:21:26 发布

阅读量453

点赞数

分类专栏：算法

本文链接：https://blog.youkuaiyun.com/edestiny1/article/details/101690631

版权

算法专栏收录该内容

6 篇文章

订阅专栏

这篇博客介绍了如何使用C++实现《算法导论》中提到的堆排序算法，包括最大堆化、建最大堆和堆排序三个关键步骤。通过示例代码详细展示了堆排序的过程，并在main函数中进行了测试。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

《算法导论》第6章堆排序
#include <bits/stdc++.h>
using namespace std;

#define PARENT(i) i/2
#define LEFT(i) 2*i
#define RIGHT(i) 2*i+1

/**
 * 最大堆化，保证子树最大值在子树的根
 * @param A 数组
 * @param len 数组长度
 * @param i 当前位置
 */
void maxHeapify(int A[], int len, int i){
    int l = LEFT(i);
    int r = RIGHT(i);
    int largest = -1;
    if (l <= len && A[l] > A[i]){
        largest = l;
    } else {
        largest = i;
    }
    if (r <= len && A[r] > A[largest]){
        largest = r;
    }

    if (largest != i){
        int temp = A[i];
        A[i] = A[largest];
        A[largest] = temp;
        maxHeapify(A, len, largest);
    }
}

/**
 * 建最大堆
 * @param A 数组
 * @param len 数组长度
 */
void buildMaxHeap(int A[], int len){
    for (int i = len/2; i >= 1; --i) {
        maxHeapify(A, len, i);
    }
}

/**
 * 堆排序
 * @param A 数组
 * @param len 数组长度
 */
void heapSort(int A[], int len){
    buildMaxHeap(A, len);
    for (int i = len; i >= 2; --i) {
        int temp = A[1];
        A[1] = A[i];
        A[i] = temp;
        len--;
        maxHeapify(A,len,1);
    }
}

int main(){
    //数组A[0]忽略，从A[1]开始
    int A[] = {0,4,1,3,2,16,9,10,14,8,7};
    int len = sizeof(A)/ sizeof(A[0]) - 1;
    heapSort(A, len);

    for (int i = 1; i <= len; ++i) {
        cout<<A[i]<<" ";
    }
    cout<<endl;
}