面试题 17.14. 最小K个数 C++

原创于 2021-09-04 22:10:40 发布 · 167 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#c++ #leetcode #算法 #数组 #排序

LeetCode刷题专栏收录该内容

47 篇文章

订阅专栏

该篇博客介绍了如何设计一个算法来找出数组中的最小k个数。首先检查数组是否为空或k是否为0，然后对数组进行排序，并将排序后的前k个元素存入结果数组中。最后返回这个包含最小k个数的结果数组。示例展示了当输入数组为[1,3,5,7,2,4,6,8]，k为4时，输出为[1,2,3,4]。这是一个基础的算法问题，主要涉及排序和数组操作。

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

题目描述

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

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/smallest-k-lcci

解题思路

首先特判：如果数组为空，或者k为0，直接返回空数组。
否则的话直接sort数组，并且把排序好的前k位添加到结果数组res中，并且返回res

代码

class Solution {
public:
    vector<int> smallestK(vector<int>& arr, int k) {
        vector<int> res;//结果数组
        if(arr.size()==0||k==0){//如果arr为空，或者k为0
            return res;//直接返回空数组
        }
        sort(arr.begin(),arr.end());//对arr进行排序
        for(int i=0;i<k;i++){//把前k个数都放到res数组中
            res.push_back(arr[i]);
        }
        return res;//返回res数组
    }
};