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