算法：堆排序（堆）

堆：

使用情形：

1. 插入一个数
2. 求集合当中的最小值
3. 删除最小值
4. 删除任意一个元素
5. 修改任意一个元素

用一维数组存储二叉树，左儿子在数组中为根节点在数组中的位置的2倍，右儿子在数组中为根节点在数组中的位置的二倍加一。

求当前堆的最小值

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

问题描述

输入一个长度为n的整数数列，从小到大输出前m小的数。

输入格式

第一行包含整数n和m。

第二行包含n个整数，表示整数数列。

输出格式

共一行，包含m个整数，表示整数数列中前m小的数。

数据范围

1≤m≤n≤105，
1≤数列中元素≤109

输入样例：

5 3
4 5 1 3 2

输出样例：

1 2 3

代码

#include<iostream>
#include<algorithm>
using namespace std;
int h[100100];
int siz = 1;
void down(int u){
    int t = u;
    if(u * 2 <= siz && h[t] >= h[u * 2]) t = u * 2;
    if(u * 2 + 1 <= siz && h[t] >= h[u * 2 + 1]) t = u * 2 + 1;
    if(u != t){
        swap(h[u], h[t]);
        down(t);
    }
}

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

原题链接