LeetCode215 数组中的第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
public int findKthLargest(int[] nums, int k) {
quickSort(nums, 0, nums.length - 1);
return nums[k - 1]; // 因为是从零开始所以k - 1
}
public static void quicksort(int[] nums, int start, int end) {
int left = start;
int right = end;
int pivot = nums[(left + right) / 2]; //取中位值作左右参考对比
//每次循环都将大的放到左边,小的放到右边
while (left < right) {
while (nums[left] > pivot) {
left++;
}
while (nums[right] < pivot) {
right--;
}
//如果l>=r说明mid左边的值全部大于等于mid的值,右边全是小的,退出
if (left >= right) {
break;
}
//交换
int temp = nums[left];
nums[left] = nums[right];
nums[right] = temp;
//解决值相同的情况,如果交换后发现左边的值等于mid目标值,则使右边指针向左移
if (nums[left] == pivot) {
right--;
}
//右边的值等于mid目标值,则使左边指针向右移
if (nums[right] == pivot) {
left++;
}
}
if (left == right) { //退出循环防止栈溢出
left++;
right--;
}
if (start < right) {
quicksort(nums, start, right); //向左递归
}
if (end > left) {
quicksort(nums, left, end); //向右递归
}
}
这里的快速排序和之前的不同,这里是选定一个pivot,左边都是比它大的数,右边都是比他小的数
这样,要输出的就是[k - 1],数组中下标从0开始。