【堆】AcWing838. 堆排序 —— 堆入门

堆

(下标一定要从0， 0无法乘2)
该讲解的是小根堆，如果是大根堆，则需转换一下思维
堆是一颗完全二叉树,且最后一层是从左往右存储的,堆每个父节点都小于等于左右结点，故根节点为最小值

堆有两个操作，其余所有操作都可由这两个操作组合而成：
down(x):当x结点的值变大的时候，我们要将x向下移动，与左右最小子节点互换，直到左右子节点都大于该点或到达最底层为止。
up(x):当x结点的值变小的时候，我们要将x向上移动，由于父节点是小于等于左右结点的，故我们只需与父节点比较，判断是否要互换，上移到父节点小于等于该点或根节点为止

AcWing838. 堆排序

题解

#include <iostream>

using namespace std;

const int N = 1e5+10;

int h[N], sz;

void down(int u)
{
    int t = u;
    if(u * 2 <= sz && h[u * 2] < h[t] ) t = u * 2;
    if(u * 2 + 1 <= sz && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
    if(t != u)
    {
        swap(h[t], h[u]);
        down(t);
    }
}

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