输入整数数组 arr ,找出其中最小的 k 个数。例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。
示例 1:
输入:arr = [3,2,1], k = 2
输出:[1,2] 或者 [2,1]
示例 2:
输入:arr = [0,1,2,1], k = 1
输出:[0]
解:
①堆排序,priority_queue优先队列结构(C++默认大根堆)(也可以自己写实现)
class Solution {
public:
//维护大小为k的优先队列(大顶堆)
//C++默认是大顶堆
priority_queue<int> a;
vector<int> res;
for(int i = 0;i < arr.size();++i)
{
if(i < k) a.push(arr[i]);
else if(arr[i] < a.top())
{
a.push(arr[i]);
a.pop();
}
//a.push(arr[i]);
//if(a.size() > k) a.pop();
}
while(!a.empty())
{
res.push_back(a.top());
a.pop();
}
return res;
/*sort(arr.begin(),arr.end());
for(int i = 0;i < k;++i)
{
res.push_back(arr[i]);
}
return res;*/
};
②快排->返回数组无排序要求->快速选择
//找到第k大的数(索引k-1)
//随机一个数,调整顺序至左边的比它小,右边的比它大
//如果该数坐标为k-1,则它就为我们要找的数
//如果该数坐标小于k-1,在该数右边的数组中继续找;如果该数坐标大于k-1,在该数左边的数组中继续找。
//未调整队列:[p,r]
//排序,取前k个
//先存k个数进res,并维护一个
class Solution {
public:
int k;
vector<int> getLeastNumbers(vector<int>& arr, int k){
if(arr.empty() || arr.size() < k || k < 1) return{};
this->k = k;
int resNum = arrAdjust(arr,0,arr.size() - 1); //得到调整数组后索引为k的位置
vector<int> res(arr.begin(),arr.begin() + resNum);
return res;
}
int arrAdjust(vector<int>& arr,int p,int r)
{
if(p > r) return p; //防止arr.size() = k时,数组越界
int val = arr[r];
int low = p;
int quick = p;
while(quick < r)
{
if(arr[quick] < val) //记得访问数组。不要直接quick。
{
swap(arr,low,quick);
low++;
}
quick++;
}
swap(arr,low,r);
if(low == k) return low;
else if(low < k) return arrAdjust(arr,low+1,r); //return 函数,不要直接调用函数
else return arrAdjust(arr,0,low-1);
}
void swap(vector<int>& arr,int v1,int v2)
{
int tmp = arr[v1];
arr[v1] = arr[v2];
arr[v2] = tmp;
}
};