Java教程

基础算法 838.堆排序

本文主要是介绍基础算法 838.堆排序,对大家解决编程问题具有一定的参考价值,需要的程序猿们随着小编来一起学习吧!
堆排序中,最主要的是这个down()函数
#include<iostream>
using namespace std;

const int N = 100010;
int h[N], cnt;

void down( int k ){
    int t = k;
    if(k * 2 <= cnt && h[k * 2] < h[t]) t = k * 2;
    if(k * 2 + 1 <= cnt && h[k * 2 + 1] < h[t]) t = k * 2 + 1;
    
    if( k != t){
        swap(h[t], h[k]);
        down(t);
    }
}

int main(){
    int n, m;
    cin>>n>>m;
    cnt = n;
    for(int i = 1; i <= n; i ++) cin >> h[i];
    
    for(int i = n / 2; i ; i --) down (i);
    
    while(m--){
        cout<<h[1]<<" ";
        h[1] = h[cnt --];
        down(1);
    }
    return 0;
}

  

这篇关于基础算法 838.堆排序的文章就介绍到这儿,希望我们推荐的文章对大家有所帮助,也希望大家多多支持为之网!