堆排序算法

基础排序算法——堆排序

堆排序是一种利用堆这个数据结构来进行排序的算法

什么是堆？

堆是利用完全二叉树来维护的一个一维数组，和线段树不同，堆不需要额外的空间。

堆这个数据结构有一个很有趣的特性，就是它的两个子节点下标分别是这个节点下标数值乘以2+1和乘以2+2

根节点为0的时候
arr[i]的子节点是arr[i*2+1]和arr[i*2+2]

根节点为1的时候
arr[i]的子节点是arr[i*2]和arr[i*2+1]

堆排序算法

堆排序分为大顶堆（最大堆）和小顶堆（最小堆）两种。

大顶堆的特点是对于一个节点，它的大小一定大于它的两个子节点。小顶堆和它恰恰相反。以下我们针对大顶堆对其进行分析。

假如我们要排序的序列是1,9,6,4,3,8,10.那么我们的堆应该是这样的

我们的遍历所有的非叶子结点，按照从右到左，从下到上的顺序。这个图中，我们遍历的下标顺序就是3,2,1。

每个节点a按照以下的步骤进行：

1.查看a节点的两个子节点，选择比父节点大的子节点b（如果两个子节点都大于父节点，则选择更大的），并把b的值赋值给a，我们只是赋值，而不交换的原因是，防止交换过后，破坏子节点和孙节点之间大小关系（假如说原本子节点大于两个孙节点好好的，一交换之后，可能出现，孙节点大于子节点了，那我们还得回去重复计算）

2.查看我们上一步选择的子节点b，它的两个子节点哪个更大，选择更大的c，将c的值赋值给b（和上一步骤相同）

3.重复以上步骤，直到我们查找到叶子结点位置结束，将我们最后一次找到的节点，赋值给a。

这样，我们成功维护一次大顶堆之后，我们不难发现，arr[0]就是数组当中第一大的元素，如果我们把它排除掉，（排除的方法是将arr[0]和arr[6]交换，并且arr[6]不参与接下来的大顶堆维护），对剩余6个元素继续维护大顶堆，我们就得到了第二大的元素。如此反复，我们就得到了这个序列从小到大的排序。

堆排序代码：

#include<stdio.h>
#include<string.h>
int arr[] = {1,9,6,4,3,8,10};
int len = 7;
void adjustHeap(int i, int len) //从i节点的子节点当中，取一个比i大的子节点与i进行交换
{
    int temp = arr[i];
    for (int k = i*2+1; k<len; k = k*2+1) // k=k*2+1，是当前k节点的左子节点
    {
        if (k + 1 < len && arr[k]<arr[k+1]) //如果右节点比左节点大，那就选择右节点
            k++;
        if (arr[k] > temp)
        {
            arr[i] = arr[k]; //不立刻交换
            i = k;
        }
        else break;
    }
    arr[i] = temp;
}
void swap(int &a,int &b) //交换函数
{
    int temp=a;
    a=b;
    b=temp;
}
void sort() //堆排序
{
    for (int i = len/2-1; i>=0; i--) //从非叶子结点开始，以从右到左，从下到上的顺序维护堆
    {
        adjustHeap(i,len);
    }

    for (int j = len-1; j>0 ;j--) //把当前第一大的元素放到最后
    {
        swap(arr[0],arr[j]);
        adjustHeap(0,j);
    }
}

int main()
{
    sort();
    for (int i=0;i<len;i++)
        printf("%d%c",arr[i],i==len-1?'\n':' ');
}

堆排序还是值得学习一下的，毕竟在C++的STL容器中有着许多应用，最为典型的就是优先队列。