算法题（11）最小的K个数

每天两道算法题

题目

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

#include <ctime>
#include <iostream>
#include <vector>
#include <algorithm>
#include <random>    /* 目的在于演示打算数组的方法 */

using namespace std;

vector<int> GetLeastNumbers_Solution(vector<int> input, int k);

int main( void )
{
    /* 建立数组并打乱 */
    vector<int> vec = {1,2,3,4,5,6,7,8,9};
    default_random_engine dre( time((time_t*)0) );
    shuffle( vec.begin(), vec.end(), dre );
    
    /* 计算最小的k个数 */
    vector<int> ret = GetLeastNumbers_Solution( vec, 4 );
    
    /* 输出打乱后的数组 */
    cout << "vec: ";
    for( vector<int>::iterator pos=vec.begin(); pos!=vec.end(); ++pos ) {
        cout << *pos << " ";
    }
    cout << endl;
    
    /* 输出最小的k个数 */
    cout << "ret: ";
    for( vector<int>::iterator pos=ret.begin(); pos!=ret.end(); ++pos ) {
        cout << *pos << " ";
    }
    cout << endl;

    return 0;
}

vector<int> GetLeastNumbers_Solution(vector<int> input, int k)
{
    // 只能填写GetLeastNumbers_Solution()函数
}

运行演示：

$ ./a.out 
vec: 1 7 5 4 3 2 8 6 9 
ret: 1 2 3 4 

C++ 代码

vector<int> GetLeastNumbers_Solution(vector<int> input, int k)
{
    /* 非法参数 k */
    if( k > input.size() ) {
        vector<int> res;
        return res;
    }
    /* 先复制，再排序，最后删除 */
    vector<int> res(input);
    sort( res.begin(), res.end() );
    res.erase( res.begin()+k, res.end() );
    return res;
}