题解 | #寻找第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]; } } };