215. 数组中的第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

方法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)

你可能感兴趣的:(215. 数组中的第K个最大元素)