刷题-力扣-面试题 17.14. 最小K个数

面试题 17.14. 最小K个数

题目链接

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/smallest-k-lcci
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

题目描述

设计一个算法，找出数组中最小的k个数。以任意顺序返回这k个数均可。

示例：

输入： arr = [1,3,5,7,2,4,6,8], k = 4
输出： [1,2,3,4]

提示：

• 0 <= len(arr) <= 100000
• 0 <= k <= min(100000, len(arr))

题目分析

1. 根据题目描述获取数组中最小的k个值
2. 对数组进行递增排序，再取前k个元素

代码

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        sort(arr.begin(), arr.end(), [] (int a, int b) -> bool { return a < b; });
        vector<int> res;
        while (k--) {
            res.emplace_back(arr[k]);
        }
        return res;
    }
};