Top K 问题全解 [C++]

最新推荐文章于 2024-05-15 17:06:22 发布

Diryl

最新推荐文章于 2024-05-15 17:06:22 发布

阅读量423

点赞数

分类专栏：算法文章标签：算法 leetcode 快速排序数据结构 c++

本文链接：https://blog.csdn.net/Diryl_G/article/details/119609795

版权

本文介绍了使用C++解决Top K问题的四种方法，包括自建堆、STL优先队列、快速排序和选择排序。详细阐述了每种方法的实现逻辑，并提供了相应代码示例。

摘要由CSDN通过智能技术生成

@算法

Top K 问题全解 [C++]

问题描述：在无序的数组中找出最大（最小）的K个数，或者是最大或最小的第K个数，或者是频次数。最简单解法是排序，对于本问题排序方法中比较好的排序方法有快排、堆排、选择排序等。
下面给出自己写的对于Top K问题的四种解法，个人认为代码逻辑较为清晰吧，所以就没写注释。（有不对的地方希望大家不吝啬指导）

代码

方法一：自建堆

#include <iostream>
#include <vector>
using namespace std;
// top K 问题全解
// 方法一：自建堆
void heapify(int* arr, int n, int i) {
   
    int child = i * 2 + 1;
    while (child < n) {
   
        if (child + 1 < n && arr[child] < arr[child+1]) {
   
            child++;
        }
        if (arr[i] > arr[child]) break;
        else {
   
            swap(arr[i], arr[child]);
            i = child;
            child = i * 2 + 1;
        }
    }
}

void buildheap(int *arr, int n) {
   
    int lastnode = n - 1;
    int parent = (lastnode - 1 ) / 2;
    for (int i = parent; i >= 0; i--) {
   
        heapify(arr, n, i);
    }
}

vector<int> heapsort(int *arr, int n, int k) {
   
    vector<int> res;
    buildheap(arr, n);
    for (int i = n - 1; i >= n - k; i--) {
   
        swap(arr[0], arr[i]);