【leecode】堆排序

堆是一颗完全二叉树。

完全二叉树是存放在一维数组中的。

这样的话，对于存放在一维数组中的完全二叉树，有一些奇妙的性质：
可以根据 下标，快速定位其 父节点、右孩子、左孩子的下标。

算法步骤参考了：https://www.bilibili.com/video/BV1fp4y1D7cj/?spm_id_from=333.337.search-card.all.click&vd_source=c1446f9805c430206dfe8579c1b485f2

讲的非常清晰。

步骤就是，先是：

• 先构造大顶堆（从 n/2 -1, 位置开始，也就是最后一个非叶子节点，往前，每个元素进行 heapfy）
• 再从后往前，遍历节点，每个节点都和 堆顶 的元素 swap，然后 heapfy 堆顶的元素。

如果还是不清晰，就直接看视频吧：

#include<iostream>
#include<vector>
using namespace std;


void heapfy(vector<int> &arr, int size, int idx)
{
    int left, right;
    int maxi;

    maxi = idx;
    left = 2 * idx + 1;
    right = 2 * idx + 2;

    if (left < size && arr[left] > arr[maxi]) {
        maxi = left;
    }
    if (right < size && arr[right] > arr[maxi]) {
        maxi = right;
    }
    if (maxi != idx) {
        swap(arr[maxi], arr[idx]);
        heapfy(arr, size, maxi);
    }
}



void heap_sort(vector<int> &arr, int size)
{
    int i = 0;
    // 先建堆
    for (i = size/2-1; i >= 0; i--) {
        heapfy(arr, size, i);
    }

    // 后排序
    for (i = size - 1; i >= 1; i--) {
        swap(arr[i], arr[0]);
        heapfy(arr, i, 0); // 注意，这里传入的长度是将未完成部分的长度
    }
}

int main()
{
    vector<int> arr = {8, 1, 14, 3, 21, 5, 7, 10};
    for(int i = 0; i < arr.size(); i++) {
        cout << arr[i] << " ";
    }
    cout << endl;
    heap_sort(arr, arr.size());
    for(int i = 0; i < arr.size(); i++) {
        cout << arr[i] << " ";
    }
    cout << endl;
    return 0;
}

mac 上编译，需要 C++ 11标准：

g++ -std=c++11 heap_sort.cpp