数组中最小的k个数
题目
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路
思路1:
利用 快排的 partition 函数对 数组进行初步排序,使得位于 第 k-1前后的数组是排序的。
输出前k个数字。
思路2:
处理海量数据时,可以使用一个k个元素的 multiset,顶上放最大的数字,不断判断是否更新set里面的数字。
实现
class Solution {
public:
int partition(vector<int>& numbers, int left, int right)
{
if(numbers.empty() || left<0 || right>=numbers.size() || left>right)
return -1;
int pivot = numbers[left];
while(left<right)
{
while(left<right && numbers[right]>=pivot)
--right;
numbers[left] = numbers[right];
while(left<right && numbers[left]<=pivot)
++left;
numbers[right] = numbers[left];
}
numbers[left] = pivot;
return left;
}
vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
vector<int> result;
if(input.empty() || k<=0 || input.size()<k)
return result;
int left = 0;
int right = input.size()-1; //记得是减1
int pindex = partition(input, left, right);
int mark = k-1;
while(pindex!=mark)
{
if(pindex<mark)
{
left = pindex+1;
pindex = partition(input, left, right);
}
else
{
right = pindex-1;
pindex = partition(input, left, right);
}
}
for(int i=0; i<k; ++i)
result.push_back(input[i]);
return result;
}
};