AcWing 838. 堆排序

Posted 2022-02-11 MangataTS

题目连接

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;