215. Kth Largest Element in an Array

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        int l = 0, r = nums.size() - 1;
        k--;
        while (l < r) {
            swap(nums[l], nums[(r - l) / 2 + l]);
            int i = l, j = r;
            int x = nums[j];
            while (i < j) {
                while (j > i && nums[j] <= x) j--;
                if (i < j) nums[i++] = nums[j];
                while (i < j && nums[i] >= x) i++;
                if (i < j) nums[j--] = nums[i];
            }
            nums[i] = x;
            if (i == k) return nums[k];
            if (i < k) l = i + 1;
                else r = i - 1;
        }
        return nums[k];
    }
};

results matching ""

    No results matching ""