一、题目
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。
二、实现
注意:寻找第k大的数,需使用快速排序对数组进行递减排序
import java.util.*;
public class Solution {
public int findKth(int[] a, int n, int K) {
return findK(a,0,n-1,K);
}
// 1.快速排序,返回已排序元素下标
public static int Quick(int[] a, int l, int h){
int p = a[l];
while(l<h){
while(l<h && a[h]<=p) h--;
if(l<h) a[l] = a[h];
while(l<h && a[l]>=p) l++;
if(l<h) a[h] = a[l];
}
a[l] = p;
return l; //返回当前已排序的元素位置
}
// 2.查找第k大元素
public static int findK(int[] a, int l, int h, int K){
if(l <= h){
int pos = Quick(a,l,h);
if(pos == K-1){
return a[pos];
}
else if(pos < K-1){ //右侧
return findK(a,pos+1,h,K);
}
else{//左侧
return findK(a,l,pos-1,K);
}
}
return -1; //查找失败
}
}