题目描述:在未排序的数组中找到第 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
方法1:快排后取K大
void quickSort(int *nums, int startIndex, int endIndex) {
if (startIndex >= endIndex) {
return;
}
int l = startIndex;
int r = endIndex;
int flagNum = nums[startIndex];
while (r > l) {
while (nums[r] > flagNum && r > l) {
r --;
}
while (nums[l] <= flagNum && r > l) {
l ++;
}
if (r > l) {
int tmp = nums[l];
nums[l] = nums[r];
nums[r] = tmp;
}
}
nums[startIndex] = nums[l];
nums[l] = flagNum;
quickSort(nums, startIndex, l - 1);
quickSort(nums, r + 1, endIndex);
}
int findKthLargest(int* nums, int numsSize, int k){
quickSort(nums, 0, numsSize - 1);
return nums[numsSize - k];
}
时间复杂度为快排的复杂度:O(N *logN),空间复杂度O(1)
方法2:对快排过程做优化,不完全排序
void quickSort(int *nums, int startIndex, int endIndex, int k, int numsSize) {
if (startIndex >= endIndex) {
return;
}
int l = startIndex;
int r = endIndex;
int flagNum = nums[startIndex];
while (r > l) {
while (nums[r] > flagNum && r > l) {
r --;
}
while (nums[l] <= flagNum && r > l) {
l ++;
}
if (r > l) {
int tmp = nums[l];
nums[l] = nums[r];
nums[r] = tmp;
}
}
nums[startIndex] = nums[l];
nums[l] = flagNum;
if (l < numsSize - k) {
quickSort(nums, r + 1, endIndex, k, numsSize);
}
else if (l > numsSize - k) {
quickSort(nums, startIndex, l - 1, k, numsSize);
}
else {
return;
}
}
int findKthLargest(int* nums, int numsSize, int k){
quickSort(nums, 0, numsSize - 1, k, numsSize);
return nums[numsSize - k];
}
时间复杂度是O(N),空间复杂度是O(1)