有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。
测试样例:
[1,3,5,2,2],5,3
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。
测试样例:
[1,3,5,2,2],5,3
返回:2
public class 寻找第K大的数 {
public static void main(String[] args) {
// TODO Auto-generated method stub
int[] a = { 1, 3, 5, 2, 2 };
System.out.println(findKth(a, a.length, 2));
/*
* qsort(a, 0, a.length-1); for(int i:a){ System.out.print(i+" "); }
*/
}
public static int findKth(int[] a, int n, int K) {
// write code here
return qs(a, 0, a.length - 1, a.length - K);
}
static void qsort(int[] a, int left, int right) {
if (left < right) {
int p = partition(a, left, right);
qsort(a, left, p - 1);
qsort(a, p + 1, right);
}
}
static int qs(int[] a, int left, int right, int k) {
// if (left < right)
{
int p = partition(a, left, right);
if (p == k) {
return a[p];
}