在未排序的数组中找到第 k 个最大的元素。请注意,你需要找的是数组排序后的第 k 个最大的元素,而不是第 k 个不同的元素。
public int findKthLargest(int[] nums, int k) {
return findKthLargest(nums, 0, nums.length,k);
}
private int findKthLargest(int[] nums, int start, int end, int k) {
if (start==end-1) {
if (k==1) return nums[start];
else return -1;
}
int n = nums[start];
int i=start+1;
int j=end-1;
while(i<=j) {
while(i<=j&&nums[i]<=n) ++i;
while(i<=j&&nums[j]>=n) --j;
if (i<j) {
int tmp = nums[i];
nums[i] = nums[j];
nums[j] = tmp;
i++;
j--;
}
}
nums[start] = nums[j];
nums[j] = n;
int right = end-j;
if (right==k) {
return n;
} else if(right<k){
return findKthLargest(nums, start, j, k-right);
} else {
return findKthLargest(nums, j+1, end, k);
}
}