堆排序

一、什么是堆？

堆是一种任意节点不大于或者不小于父节点的完全二叉树结构。

由堆的定义可以知道堆的父节点一定大于或者小于其孩子节点，同时每个结点的左子树和右子树都是一个二叉堆（都是最大堆或最小堆）。其中根节点最大的堆叫做最大堆或大根堆，根节点最小的堆叫做最小堆或小根堆。

二、堆排序基本思想

1) 建立最大堆（最小堆），此时堆顶元素是该组数据中最大数据（最小数据）

2) 移除堆顶元素，重新建立最大堆（最小堆）

3) 重复步骤2)

4) 数据全部移除，移除数据构建成有序序列

三、示例分析 (arr[] = {8, 6, 1, 4, 0, 3, 9, 5, 7, 2})

       

    

    

  

四、代码实现

#include <iostream>
#include <assert.h>
using namespace std;

void AdjustDown(int* arr, int root, int n)
{
    int parent = root;
    int child = parent * 2 + 1;
    while (child < n) { // 选孩子里面大的一个
        if (child + 1 < n && arr[child + 1] > arr[child]) {
            ++child;
        }
        if (arr[parent] < arr[child]) {
            swap(arr[parent], arr[child]);
            parent = child;
            child = parent * 2 + 1;
        } else {
            break;
        }
    }
}

void HeapSort(int* arr,int size)
{
    assert(arr);
    // 建大堆
    for(int i = (size-2)/2; i >= 0; --i) {
        AdjustDown(arr, i, size);
    }
    // 排序
    int end = size-1;
    while (end > 0) {
        swap(arr[0], arr[end]);
        AdjustDown(arr, 0, end);
        --end;
    }
}

int main(int argc,const char* argv[])
{
    int arr[] = {8, 6, 1, 4, 0, 3, 9, 5, 7, 2};
    HeapSort(arr, sizeof(arr)/sizeof(arr[0]));
    for (int i = 0; i < sizeof(arr)/sizeof(arr[0]); i++) {
        cout<<arr[i]<<" ";
    }
    cout<<endl;
    return 0;
}