在未排序的数组中找到第 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
说明:你可以假设 k 总是有效的,且 1 ≤ k ≤ 数组的长度。
解法1:先排序再查找
public int findKthLargest(int[] nums, int k) {
Arrays.sort(nums);
return nums[nums.length - k];
}
解法2:使用快排的思想
核心思想是每次都要先找一个中枢点Pivot,然后遍历其他所有的数字,像这道题从小往大排的话,就把小于中枢点的数字放到左半边,把大于中枢点的放在右半边,这样中枢点是整个数组中第几大的数字就确定了,虽然左右两部分不一定是完全有序的,但是并不影响本题要求的结果,所以我们求出中枢点的位置,如果正好是k-1,那么直接返回该位置上的数字;如果大于k-1,说明要求的数字在左半部分,更新右边界,再求新的中枢点位置;反之则更新右半部分,求中枢点的位置。
public int findKthLargest(int[] nums, int k) {
int left = 0;
int right = nums.length - 1;
while(true){
int pos = partation(nums, left, right);
if(pos == k - 1) return nums[pos];
if(pos > k - 1){
right = pos - 1;
}else{
left = pos + 1;
}
}
}
public int partation(int[] nums, int left, int right){
int privote = nums[left], l = left + 1, r = right;
while(l <= r){
if(nums[l] < privote && nums[r] > privote){
swap(nums, l++, r--);
}
if(nums[l] >= privote) l++;
if(nums[r] <= privote) r--;
}
swap(nums, left, r);
return r;
}
private void swap(int[] nums, int i, int j){
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
}