其他分享
首页 > 其他分享> > 【堆】AcWing838. 堆排序 —— 堆入门

【堆】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;
}

标签:结点,入门,int,堆排序,AcWing838,down,根堆,节点
来源: https://www.cnblogs.com/czy-algorithm/p/16293106.html