求最小的k个数

最新推荐文章于 2024-08-27 14:41:49 发布

青石玄霄

最新推荐文章于 2024-08-27 14:41:49 发布

阅读量125

点赞数

分类专栏：算法数据结构 C++

本文链接：https://blog.csdn.net/u014603046/article/details/117093762

版权

C++ 同时被 3 个专栏收录

25 篇文章 0 订阅

订阅专栏

算法

9 篇文章 0 订阅

订阅专栏

数据结构

3 篇文章 0 订阅

订阅专栏

本文探讨了如何使用排序和堆排序两种方法，分别实现找出给定数组中最小的K个数。排序法通过直接对数组进行排序，而堆排序则利用优先队列高效地找到并维护K个最小元素。适合初学者理解这两种经典算法在实际问题中的应用。

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

给定一个数组，找出其中最小的K个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。如果K>数组的长度，那么返回一个空的数组

方法一：排序

直接排序

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> ret;
        if (k==0 || k>input.size()) return ret;
        sort(input.begin(), input.end());
        return vector<int>(input.begin(), input.begin()+k);   
    }
};

方法二：堆排序

建立一个容量为k的大根堆（默认就是）的优先队列。遍历一遍元素，如果队列大小<k,就直接入队，否则，让当前元素与队顶元素相比，如果队顶元素大，则出队，将当前元素入队

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        vector<int> ret;
        if (k==0 || k > input.size()) return ret;
        priority_queue<int, vector<int>> pq;
        for (const int val : input) {
            if (pq.size() < k) {
                pq.push(val);
            }
            else {
                if (val < pq.top()) {
                    pq.pop();
                    pq.push(val);
                }

            }
        }

        while (!pq.empty()) {
            ret.push_back(pq.top());
            pq.pop();
        }
        return ret;
    }
};