题解 | #寻找第K大#
寻找第K大
https://www.nowcoder.com/practice/e016ad9b7f0b45048c58a9f27ba618bf
#include <cstdlib>
#include <vector>
class Solution {
public:
/**
* 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
*
*
* @param a int整型vector
* @param n int整型
* @param K int整型
* @return int整型
*/
int findKth(vector<int>& a, int n, int K) {
// write code here
return quicksort(a, 0, n - 1, K);
}
int quicksort(vector<int>&nums,int left,int right,int k){
if(left>right) return -1;
int i=left,j=right;
int pivot=nums[left];
while(i<j){
while(i<j&&nums[j]<=pivot) j--;
while(i<j&&nums[i]>=pivot) i++;
if(i<j){
swap(nums[i],nums[j]);
}
}
swap(nums[left],nums[j]);
if(j<k-1){//左边元素不足k 在右半部分找
return quicksort(nums, j+1, right, k);
}else if(j>k-1){//右边元素不足k 在左半部分找
return quicksort(nums, left, j-1, k);
}else{
return nums[j];
}
}
};
