求前 k 大的那一串

👁️ 0 人浏览 💬 0 人评论 ❤️ 添加收藏

补全 top_k：返回前 k 大组成的数组（从大到小）。这次求前 3 大，用 / 拼起来输出。

（本题用 g++ -std=c++17 -O0 编译。）

提交你的答案

请登录后提交答案。

去登录

← 求第 k 大大顶堆里最小的在哪 →

更多题目

代码编辑器

语言： Ctrl + Enter 运行

#include <algorithm>
#include <iostream>
#include <string>
#include <utility>
#include <vector>
using namespace std;

string join(const vector<int>& v) {
    string s;
    for (size_t i = 0; i < v.size(); i++) s += (i ? "/" : "") + to_string(v[i]);
    return s;
}

void sift_down(vector<int>& h, int i, int n) {
    while (true) {
        int big = i;
        int l = 2 * i + 1;
        int r = 2 * i + 2;
        if (l < n && h[l] > h[big]) big = l;
        if (r < n && h[r] > h[big]) big = r;
        if (big == i) return;
        swap(h[i], h[big]);
        i = big;
    }
}

int pop_top(vector<int>& h) {
    int top = h[0];
    int last = h.back();
    h.pop_back();
    if (!h.empty()) {
        h[0] = last;
        sift_down(h, 0, (int)h.size());
    }
    return top;
}

void build_heap(vector<int>& h) {
    int n = (int)h.size();
    for (int i = n / 2 - 1; i >= 0; i--) sift_down(h, i, n);
}

vector<int> top_k(vector<int> vals, int k) {
    vector<int> out;
    // TODO: 建堆、连弹 k 次，把每次弹出的收进 out
    return out;
}

int main() {
    cout << join(top_k({17, 24, 15, 13, 23}, 3)) << endl;
}

本次输入：

输出：

👩‍🏫

💬 题目评论

求前 k 大的那一串

更多题目

全部评论