C++：最小的K个数

题目描述

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

示例1

输入

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

返回值

[1,2,3,4]

 

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {

        vector<int> result;

        if(input.size() < k)
        {
            return result;
        }

        for(int i = 0; i < k; i++)
        {
            if(input.empty())
            {
                return result;
            }

            int val = input[0];

            auto input_iter_record = input.begin();
            for(auto input_iter = input.begin(); input_iter < input.end(); input_iter++)
            {
                if(*input_iter < val)
                {
                    val = *input_iter;
                    input_iter_record = input_iter;
                }
            }

            result.push_back(val);
            input.erase(input_iter_record);
        }

        return result;
    }
};