【算法练习】最小的k个数_最小的k个数的平均值-优快云博客

该问题旨在找出数组中最小的k个不重复的数字。提供的解决方案包括对数组进行排序以及使用优先队列（堆）的方法。排序方法直接选取前k个最小元素，而优先队列方法则能在线性时间复杂度内得到结果。两种方法都满足了空间复杂度O(n)和时间复杂度O(nlogk)的要求。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

描述

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围：0≤k,n≤10000，数组中每个数的大小0≤val≤1000

要求：空间复杂度 O(n) ，时间复杂度 O(nlogk)

示例1

输入：

[4,5,1,6,2,7,3,8],4

返回值：

[1,2,3,4]

说明：

返回最小的4个数即可，返回[1,3,2,4]也可以

示例2

输入：

[1],0

返回值：

[]

示例3

输入：

[0,1,2,1,2],3

返回值：

[0,1,1]

思路：

排序后放回答案数组即可，题解2用优先队列

题解：

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        sort(input.begin(),input.end());
        vector<int> ans;
        for(int i = 0;i < k;i++){
            ans.push_back(input[i]);
        }
        return ans;
    }
};

//优先队列
class Solution {
  public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
        priority_queue<int, vector<int>, greater<int>> ss;
        vector<int> s;
        for (int i = 0; i < input.size(); i++) {
            ss.push(input[i]);
        }
        for (int i = 0; i < k; i++) {
            s.push_back(ss.top());
            ss.pop();
        }
        return s;
    }
};