题目描述
有一个整数数组,请你根据快速排序的思路,找出数组中第K大的数。
给定一个整数数组a,同时给定它的大小n和要找的K(K在1到n之间),请返回第K大的数,保证答案存在。
测试样例:
[1,3,5,2,2],5,3
返回:2
题解:
利用快速排序的方法,找准第一个数的最终位置,下标为index。如果它的右侧数量正好是k-1个,那么它就是第k大的数,因为比它大的数最终都会在它的右边;如果右侧比k-1小,则要找的数在左边,递归求解左边第k-(右侧个数+1)
package com.wx.nowcoder;
public class NC88 {
public int findKth(int[] a, int n, int K) {
return find(a, 0, n - 1, K);
}
private int find(int[] a, int left, int right, int k) {
int index = partion(a, left, right);
if (right - index + 1 == k) {
return a[index];
}
if (right - index + 1 < k) {
return find(a, left, index - 1, k - right + index - 1);
}
return find(a, index + 1, right, k);
}
private int partion(int[] arr, int left, int right) {
int key = arr[left];
while (left < right) {
while (left < right && arr[right] >= key) {
right--;
}
arr[left] = arr[right];
while (left < right && arr[left] <= key) {
left++;
}
arr[right] = arr[left];
}
arr[left] = key;
return left;
}
public static void main(String[] args) {
NC88 nc = new NC88();
int[] a = {4, 1, 3, 5, 2};
int k = 3;
int i = nc.findKth(a, a.length, k);
System.out.println(i);
}
}