class Solution {
public:
void quick_sort(vector<int>& arr, int left, int right){=
if(left>=right || left<0 || right>arr.size()-1)
return;
int i=left, j=right, temp=0;;
int index=arr[left];//每次都以区间内的第一个数为标准,找到它在整个数组中的正确位置
while(i<j){
while(i<j && index<=arr[j]) //从右到左找到第一个满足条件的数
j--;
if(i==j)
break;
arr[i]=arr[j];
while(i<j && index>=arr[i]) //从左到右找到第一个满足条件的数
i++;
if(i==j)
break;
arr[j]=arr[i];
}
arr[i]=index;
quick_sort(arr, left, i-1);
quick_sort(arr, i+1, right);
}
vector<int> getLeastNumbers(vector<int>& arr, int k) {
if(arr.size()==0)
return arr;
vector<int> res;
if(k==0)
return res;
quick_sort(arr, 0, arr.size()-1);
res.assign(arr.begin(), arr.begin()+k);
return res;
}
};
【排序】用快速排序解决TopK问题
最新推荐文章于 2024-09-04 09:05:44 发布