简单算法 寻找第K大(java)
描述
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(1<=K<=n),请返回第K大的数(包括重复的元素,不用去重),保证答案存在。
public int findKth(int[] a, int n, int K) {
return quickSort(a, 0, n-1, K);
}
public int quickSort(int[] a, int left, int right, int k) {
int flag = a[left];
int l = left, r = right;
while(l < r){
while(l < r && flag >= a[r]) r--;
a[l] = a[r];
while(l < r && flag <= a[l]) l++;
a[r] = a[l];
}
a[l] = flag;
if(l == k - 1){
return a[l];
}else if(l > k - 1){
return quickSort(a,left,l - 1,k);
}else {
return quickSort(a,l + 1,right,k);
}
}