0
点赞
收藏
分享

微信扫一扫

AcWing 838. 堆排序

才德的女子 2022-01-26 阅读 49

题目连接

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;
}
举报

相关推荐

0 条评论