LintCode(5) - k 番目に大きい要素

配列内で k 番目に大きい要素を検索します。ps: クイックソートを使用してください

class Solution {
public:
    /**
     * @param k: An integer
     * @param nums: An array
     * @return: the Kth largest element
     */
    int kthLargestElement(int k, vector<int> &nums) {
        // write your code here
        int n = nums.size();
        return partition(nums, 0, n-1, k);
    }
    int partition(vector<int>& nums, int start, int end, int k) {
        int left = start, right = end;
        int pivot = nums[left];
        
        while (left <= right) {
            while (left <= right and nums[left] > pivot) {
                left++;
            }
            while (left <= right and nums[right] < pivot) {
                right--;
            }
            if (left <= right) {
                swap(nums[left], nums[right]);
                left++;
                right--;
            }
        }
        if ((k-1) <= right) {
            return partition(nums, start, right, k);
        }
        if ((k-1) >= left) {
            return partition(nums, left, end, k);
        }
        return nums[k-1];
    }
};