// 面试题40:最小的k个数
// 题目:输入n个整数,找出其中最小的k个数。例如输入4、5、1、6、2、7、3、8
// 这8个数字,则最小的4个数字是1、2、3、4。
// ====================方法1====================
static bool get_least_knum(std::vector<int>& arr, std::vector<int>& out, uint32_t k)
{
if(k < 1 || arr.size()<k)
{
return false;
}
size_t start = 0;
size_t end = arr.size() - 1;
size_t index = _partition(arr, start, end);
while(index != k - 1)
{
if(index > k - 1)
{
end = index - 1;
index = _partition(arr, start, end);
}
else
{
start = index + 1;
index = _partition(arr, start, end);
}
}
out.resize(k);
for(uint32_t i = 0; i < k; ++i)
{
out[i] = arr[i];
}
return true;
}