【leectode 2021.12.29】数组中的第 k 大的数字

给定整数数组 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);
        }
    }
}