AcWing 838. 堆排序

AcWing 838. 堆排序
输入一个长度为 n 的整数数列$a[i]$，从小到大输出前 m 小的数。

输入格式

第一行包含整数 n 和 m。

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

输出格式

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

数据范围

$1\le m\le n\le 10^5，1\le a[i]\le 10^9.$

输入样例：

5 3
4 5 1 3 2

输出样例：

1 2 3

#include<bits/stdc++.h>
#include <unordered_map>
using namespace std;
template <typename T> void debug(string s, T x) { cout << s << "=" << x << "\n"; }
typedef long long ll;
typedef unsigned long long ull;
typedef pair<ll, ll>pll;
typedef pair<int, int>pii;
const ll N = 1e6 + 5;
const ll MOD = 1e9+7;
const ll INF = 0x7fffffff;
//堆是具有以下性质的完全二叉树：
//每个结点的值都大于或等于其左右孩子结点的值，称为大顶堆；
//或者每个结点的值都小于或等于其左右孩子结点的值，称为小顶堆
int heap[N], _size;
void down(int u) {
    int t = u;//t中存储三个点(节点和左右儿子)中的最小值
    int ls = u << 1, rs = u << 1 | 1;
    //判断最小值位置，由于堆的性质先判断左儿子的大小
    if (ls <= _size && heap[ls] < heap[t])t = ls;
    if (rs <= _size && heap[rs] < heap[t])t = rs;
    if (u != t) {//如果最小值不是此时的根节点那么要将最小值移动到根节点
        swap(heap[u], heap[t]);
        down(t);//递归下放换了位置后的节点
    }
}

int main() {
    ios_base::sync_with_stdio(false); cin.tie(0); cout.tie(0);
    
    int n, m;
    cin >> n >> m;
    for (int i = 1; i <= n; i++)cin >> heap[i];//随机的建立堆
    _size = n;
    for (int i = n / 2; i; i--)down(i);//从最后的不是叶节点开始向上下放每个个节点,O(n)时间复杂度
    while (m--) {
        cout << heap[1]<<" ";
        heap[1] = heap[_size];//与最后一个元素换位置
        _size--;//大小减少一位，即删除了首元素
        down(1);//将最后一个元素换位后下放
    }

    return 0;
}

void heapify(int heap[],int size,int u) {//大顶堆
    //选出三个中最小的 
    int maxv =u;
    int ls = u * 2 + 1, rs = u * 2 + 2;//从0开始建立的堆
    if (ls < size && heap[maxv] < heap[ls])maxv = ls;//最大值不存在不取ls<=size
    if (rs < size && heap[maxv] < heap[rs])maxv = rs;
    //如果最大的哪个不根节点
    if (u != maxv) {
        swap(heap[u], heap[maxv]);//最大的节点与根节点交换
        heapify(heap, size, maxv);//交换后的子树需要调整
    }
}
void HeapSort(int heap[],int n) {
    //从一棵完整的二叉树开始，我们可以通过在堆的所有非叶元素上运行heapify函数来将其修改为大顶堆
    for (int i = n / 2 - 1; i >= 0; i--)heapify(heap, n, i);
    for (int i = n - 1; i > 0; i--) {
        swap(heap[0], heap[i]);//第一个与最后一个元素交换
        //调整范围减小
        heapify(heap, i, 0);
    }
}