C++&Python描述 LeetCode 面试题 17.14. 最小K个数

Posted 2021-09-18 亓官劼

C++&Python描述 LeetCode 面试题 17.14. 最小K个数

  大家好，我是亓官劼（qí guān jié ），在公众号、CSDN、GitHub、B站、华为开发者论坛等平台分享一些技术博文，全平台的id皆为：亓官劼（qí guān jié ），除以上平台，其他的亓官劼id都不是本人，有不少网站在冒充，注意甄别。
放弃不难，但坚持一定很酷！时光荏苒，未来可期，一起加油~

  建了个小交流群，Q群：545611263，要进微信群的话可以加我V:qiguanjie2015备注拉群即可。近期对博文更新策略做了些调整，微信公众号后期将用来更新一些总结性的文章（更新频率将会很低），类似于刷题、一些bug解决等杂项记录的blog就更新在CSDN、华为开发者论坛和GitHub上。

  同时文章在GitHub中进行了开源，内含本系列文章目前已刷的各个题解和解题思路，GitHub地址为：LeetCode，如果文章对你有帮助的话可以来GitHub点个star，如果有更好的解题思路的话，也可以来GitHub提交~一起改进

题目

设计一个算法，找出数组中最小的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))

解题思路

排序后取前K个，或者也可以用最大堆做。

C++实现

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

Python实现

class Solution:
    def smallestK(self, arr: List[int], k: int) -> List[int]:
        arr.sort()
        return arr[:k]