《剑指offer》最小的K个数
一、题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,
二、输入描述
输入给整数数组,和K
三、输出描述
找出其中最小的K个数
四、牛客网提供的框架
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
}
};
五、解题思路
1)先用插入排序对数组中数进行排序。
2)取出有序数组中最小的kge。
附:这只是一个比较简单的方法。如果考虑效率的话,建议使用堆排序。
六、代码
#include<iostream>
#include<vector>
using namespace std;
class Solution {
public:
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> vt = insertSort(input);
vector<int> result;
if(input.size() < k) return result;
for(int i = 0; i < k; i++)
{
result.push_back(vt[i]);
}
return result;
}
vector<int> insertSort(vector<int> input)
{
for(int i = 1; i < input.size(); i++)
{
int temp = input[i];
int j = i - 1;
while(j >= 0 && input[j] > temp)
{
input[j+1] = input[j];
j--;
}
input[j+1] = temp;
}
return input;
}
};