题目连接
https://www.acwing.com/problem/content/description/840/
思路
使用堆排序的时候我们只需要实现简单的up操作(向上更新)和down操作(向下更新),然后每次输出1位置的数,然后更新即可
代码
#include<bits/stdc++.h>
using namespace std;
//----------------自定义部分----------------
#define ll long long
#define mod 1000000009
#define endl "\n"
#define PII pair<int,int>
int dx[4]={0,-1,0,1},dy[4]={-1,0,1,0};
ll ksm(ll a,ll b) {
ll ans = 1;
for(;b;b>>=1LL) {
if(b & 1) ans = ans * a % mod;
a = a * a % mod;
}
return ans;
}
ll lowbit(ll x){return -x & x;}
const int N = 2e6+10;
//----------------自定义部分----------------
int n,m,q,h[N],cnt;
void down(int u){
int t = u;
if(u * 2 <= cnt && h[u * 2] < h[t]) t = u * 2;
if(u * 2 + 1 <= cnt && h[u * 2 + 1] < h[t]) t = u * 2 + 1;
if(u != t) {
swap(h[u],h[t]);
down(t);
}
}
void up(int u){
while((u >> 1) && h[u >> 1] > h[u]){
swap(h[u>>1],h[u]);
u>>=1;
}
}
int main()
{
std::ios::sync_with_stdio(false);
std::cin.tie(nullptr);
std::cout.tie(nullptr);
cin>>n>>m;
for(int i = 1;i <= n; ++i) cin>>h[i];
cnt = n;
for(int i = n;i >= 1; -- i) down(i);
while(m--){
cout<<h[1]<<" ";
h[1] = h[cnt];
cnt--;
down(1);
up(1);
}
return 0;
}