215. 数组中的第K个最大元素 中等 快速选择算法+随机选取基准元素

215. 数组中的第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 <= 104
-104 <= nums[i] <= 104

平均时间复杂度是O(n)

注意这里快速选择选择的是下标，可以方便写递归程序

import java.util.Random;

class Solution {
    Random random = new Random();

    public int findKthLargest(int[] nums, int k) {
        return quickselect(nums,0,nums.length-1,nums.length-k);
    }

    int quickselect(int nums[],int l,int r ,int index){
    	int t=partition(nums,l,r);
    	if(t==index)return nums[t];
    	return t<index?quickselect(nums,t+1,r,index):quickselect(nums,l,t-1,index);
    }

    int partition(int nums[],int l,int r){
    	int rd=random.nextInt(r-l+1)+l;
    	swap(nums,rd,r);
    	int x=nums[r];
    	int i=l-1;
    	for(int j=l;j<r;++j){
    		if(nums[j]<x)swap(nums,++i,j);
    	}
    	swap(nums,++i,r);
    	return i;
    }
    
    void swap(int nums[],int l,int r){
    	int t=nums[l];
    	nums[l]=nums[r];
    	nums[r]=t;
    }
}