给定整数数组 nums 和整数 k,请返回数组中第 k 个最大的元素。
请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
示例 1:
输入: [3,2,1,5,6,4] 和 k = 2
输出: 5
示例 2:
输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4
提示:
1 <= k <= nums.length <= 10^4
-10^4 <= nums[i] <= 10^4
java代码:
class Solution {
public int findKthLargest(int[] nums, int k) {
return findKth(nums, 0, nums.length-1, k-1);
}
private int positition(int[] arr, int l, int r){
//随机选取标定点
int index = (int)(Math.random()*(r-l+1)+l);
int temp = arr[index];
arr[index] = arr[l];
arr[l] = temp;
int v = arr[l];
//双路快排
int i=l+1, j=r;
while (true){ //while判断为真, 可以保证即使只有两个元素时,即i=l+1=j 时,也可以进入循环
while (i<=r && arr[i]>v){ //i停下的位置一定是小于或等于v的位置
i++;
}
while (j>=l+1 && arr[j]<v){ //j停下的位置一定是大于或等于v的位置
j--;
}
if(i>j){ //只能取大于
break;
}
int temp0 = arr[i];
arr[i]=arr[j];
arr[j] = temp0;
i++;
j--;
}
int temp1 = arr[l];
arr[l] = arr[j];
arr[j]= temp1; //j落在最后一个大于等于v的位置, i落在第一个小于等于v的位置,最终i不等于j, 而l处需要一个大于v的值,所以只能和 j 处的交换
return j;
}
private int findKth(int[] arr, int l, int r, int k){
if(l==r){
return arr[l];
}
int p = positition(arr, l, r);
if(k==p){
return arr[p];
}else if(k<p){
return findKth(arr, l, p-1, k);
}else{
return findKth(arr, p+1, r, k);
}
}
}