每日一题(2-数组中的第K个最大元素)

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

提示：

1 <= k <= nums.length <= 104
-104 <= nums[i] <= 104

链接：leetcode-cn.com/problems/kt…

解题思路

快排 利用快速排序的思路，每次随机选择一个数n将数组分为两部分，左边的数大于右边的数，根据n的下标是否大于k判断继续递归左边还是右边，直到n的下标为k+1。

代码

class Solution {
public:
    int QuickSort(vector<int> &nums, int l, int r, int k){
        int key = nums[l];
        int st = l, ed = r;
        while (st < ed){
            while (nums[ed]<key&&ed>st){
                ed--;
            }
            nums[st]=nums[ed];
            while (nums[st]>=key&&ed>st){
                st++;
            }
            nums[ed]=nums[st];
        }
        nums[ed]=key;
        if(ed==k)return nums[ed];
        if(ed<k)return QuickSort(nums, ed + 1, r, k);
        return QuickSort(nums, l, ed - 1, k);
    }
    int findKthLargest(vector<int> &nums, int k){
        return QuickSort(nums, 0, nums.size() - 1, k-1);
    }
};