AcWing 838. 堆排序
原题链接
简单
作者:
鹰
,
2021-03-28 23:01:49
,
所有人可见
,
阅读 323
#include<iostream>
#include<stdio.h>
#include<algorithm>
using namespace std;
const int N = 100010;
int h[N]; //用一维数组来存二叉树,i节点左儿子就是2i,右儿子是2i+1
int n, m, size1;
void down(int u) //找三个元素组成的小堆中最小的数然后和堆顶u交换,交换后的再从t往下维护堆
{
int t = u;
if(u * 2 <= size1 && h[t] > h[u * 2]) t = u * 2; //当要访问的数据没超过堆的范围
if(u * 2 + 1 <= size1 && h[t] > h[u * 2 + 1]) t = u * 2 + 1;
if(u != t)
{
swap(h[u], h[t]);
down(t);
}
}
int main()
{
scanf("%d%d",&n,&m);
for(int i = 1; i <= n; i ++)
{
scanf("%d", &h[i]);
}
size1 = n;
for(int i = n / 2; i > 0; i --) //从完全二叉树的倒数第二排最右边一个元素开始从后往前一个一个来维护堆,该操作为O(n)
{
down(i);
}
while(m --)
{
cout << h[1] << " "; //输出堆顶即最小
h[1] = h[size1]; //把输出过的元素拿堆尾元素覆盖掉
down(1); //从1处开始向下维护堆
size1 --; //把堆尾删掉,就是刚刚输出过的
}
return 0;
}