社区讨论

堆排序,20分,求调!

P1177【模板】排序参与者 2已保存回复 10

讨论操作

快速查看讨论及其快照的属性,并进行相关操作。

当前回复
10 条
当前快照
1 份
快照标识符
@lz9tpg80
此快照首次捕获于
2024/07/31 20:28
2 年前
此快照最后确认于
2024/07/31 21:59
2 年前
查看原帖
(我记得堆排序复杂度的确是 O(nlogn)O(n\log n) 啊)
CPP
#include <iostream>
#include <vector>

using namespace std;

vector<int> a;

int n;

void rebuild(int p, vector<int>& arr, int len) {
    int val = arr[p];
    int mx = p, ls = p * 2, rs = p * 2 + 1;
    if (ls <= len && arr[ls] > arr[mx]) mx = ls;
    if (rs <= len && arr[rs] > arr[mx]) mx = rs;
    if (mx != p) {
        swap(arr[p], arr[mx]);
        rebuild(mx, arr, len);
    }
}

void heapsort(vector<int>& arr) {
    int len = arr.size() - 1;
    for (int i = (len - 1) >> 1; i; i--)
        rebuild(i, arr, len);
    for (int i = len; ~(-i); i--) {
        swap(arr[1], arr[i]);
        rebuild(1, arr, len);
    }
}

int main() {
    cin >> n;
    a.push_back((int)NULL);
    for (int i = 1, tmp; i <= n; i++) {
        cin >> tmp;
        a.push_back(tmp);
    } heapsort(a);
    for (int i = 1; i <= n; i++)
        cout << a[i] << ' ';
    return 0;
}

回复

10 条回复,欢迎继续交流。

正在加载回复...