剑指Offer II 076、数组中的第K大的数字
官方题解用的也是快速排序,只关心符合条件的一半,但是有很多方法。并且我觉得官方题解的快速排序和我以前学的快速排序的写法不一样,有点点难理解,所以使用正常的快速排序的方法来解题,下面是我写的代码。
class Solution {
int result = 0;
Random random = new Random();
public int findKthLargest(int[] nums, int k) {
int len = nums.length;
int target = len - k; // 要找的目的下标。
quickSort(nums, 0, len - 1, target);
return result;
}
public void quickSort(int[] nums, int left, int right, int target) {
if(left > right) return; // 如果左边界大于有边界,不符合条件,直接返回。
/**
如果数组完全有序,对数组进行快速排序的话,时间复杂度很高,所以
使用随机数来打乱当前的数组。
**/
int k = random.nextInt(right - left + 1) + left;
swap(nums, k, right);
/**
下面是快速排序的代码,首先找到一个pivot,这里pivot = nums[left],
快速排序,i必须小于j,使用while循环,先从后向前遍历,比较当前
元素和pivot的关系,如果大于等于pivot就j--;
然后当i>=j或者当前位置的元素小于pivot,就要把小于pivot的元素
移动到左边;
遍历i的思路是一样的。
最后把i位置设置为pivot,这也就是pivot在数组里面的正确位置。
**/
int i = left, j = right, pivot = nums[left];
while(i < j) {
while(i < j && nums[j] >= pivot) {
j--;
}
nums[i] = nums[j];
while(i < j && nums[i] <= pivot) {
i++;
}
nums[j] = nums[i];
}
nums[i] = pivot;
/**
如果当前元素就是目标元素,赋值给result,直接返回。
**/
if(i == target) {
result = nums[i];
return;
}else{
/**
如果不是的话,只需要考虑符合条件的一半,如果target在i的左边,
就对左边进行排序;
如果target在i的右边,就对右边进行排序。
**/
if(i - 1 >= left && i > target && left <= target) {
quickSort(nums, left, i - 1, target);
}
if(i + 1 <= right && i < target && right >= target) {
quickSort(nums, i + 1, right, target);
}
}
}
public void swap(int[] nums, int i, int j) {
int temp = nums[i];
nums[i] = nums[j];
nums[j] = temp;
}
}