【时间效率】最小的K个数

此题出自牛客网的剑指offer专题

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,

解题思路

由于需要找出数组中最小的K个数，所以我们可以先对这个数组进行排序。然后循环添加下标从0开始到下标为（k-1）的数字至我们定义的结果集中即可。当然，在添加之前我们需要进行判断输入数字的合法性，倘若k值大于数组长度，数组长度为0，k值为0，对于这些情况我们应当直接返回空的结果集而无需进行下面的排序，添加操作。

代码如下

Java版本

import java.util.ArrayList;
import java.util.Arrays;

public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> output = new ArrayList<>();
        if(input.length==0||k==0||input.length<k){
            return output;
        }
        Arrays.sort(input);
        for(int i=0;i<k;i++){
            output.add(input[i]);
        }
        return output;
    }
}

C++版本

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