【算法基础】堆排序

小顶堆：父节点保存的值小于儿子结点保存的值

算法描述：对于一个数组，把这个数组看成一个树的结构的话，那么对于下标为i的结点，它的儿子就应该是2*i或者2*i+1（如果存在的话）。我们要进行堆排序，就必须从下到上，将儿子结点和父节点相比较，交换，由于每次交换会涉及到子树的结构，所以又要从上至下将儿子结点和父节点相比较，交换。这里的比较，交换就是下面代码中的down函数。以下是具体的代码实现。

#include<iostream>

using namespace std;
const int N = 1e5+10;
int a[N];
int n,m;

void down(int i)
{
    int t=i;
    if(2*i<=n&&a[2*i]<a[t])t=2*i;
    if(2*i+1<=n&&a[2*i+1]<a[t])t=2*i+1;
    if(i!=t)
    {
        swap(a[i],a[t]);
        down(t);
    }
}

int main()
{
    scanf("%d%d",&n,&m);
    for(int i=1;i<=n;i++)scanf("%d",&a[i]);
    
    for(int i=n/2;i>=1;i--)down(i);
    
    for(int i=1;i<=m;i++)
    {
        cout<<a[1]<<" ";
        a[1]=a[n--];
        down(1);
    }
    
    return 0;
}

由于堆排序不是将数组整个有序，我们要获取前m个小的数必须输出一个之后让另外一个数占据顶部的位置继续执行down操作继续输出

大致的过程可以参考以下图：

最初的结构

 

 

  经典习题：堆排序

图片来源