思路一:暴力破解,无序数组排序,排完序之后找到第K个最大的数,排序方法有多种,但是无序数组排序快排时间复杂度比较低是O(nlogn)
思路二:快速选择
快速选择是在快速排序的基础上进行的优化,快速选择是把找到的基准点与要找的第K个数的下标做比较,如果正好是第K个最大的数直接返回即可,如果是比基准点大则递归右侧,小的话则递归左侧。与快速排序相比可以少递归一半的数据。
Random random = new Random();
public int findKthLargest(int[] nums, int k) {
return quickSelect(nums, nums.length - k, 0, nums.length - 1);
}
private int quickSelect(int[]nums,int idx,int l,int r){
int q = random.nextInt(r - l + 1) + l;
swap(nums, q, r);
int point=partition(nums, l, r);
if (point == idx) {
return nums[point];
}
return point > idx ? quickSelect(nums, idx, l, point - 1) : quickSelect(nums, idx, point + 1, r);
}
private int partition(int[] nums, int l, int r) {
int x=nums[r],i=l-1;
for (int j = l; j < r; j++) {
if (nums[j] <= x) {
swap(nums,++i,j);
}
}
swap(nums,i+1,r);
return i + 1;
}
private void swap(int[]nums,int l,int r){
if (l == r) {
return;
}
int temp = nums[l];
nums[l] = nums[r];
nums[r] = temp;
}