寻找数组中第k大的元素

KthLargestNumber.cpp

int findKthLargest(vector<int>& nums, int k) {
    int lo = 0;
    int high = (int)nums.size() - 1;

    for(; lo < high; ) {
        int i = lo, j = high;
        int pivot = nums[lo];
        while(i < j) {
            while(i < j && pivot >= nums[j]) {
                j--;
            }
            nums[i] = nums[j];
            while(i < j && pivot <= nums[i]) {
                i++;
            }
            nums[j] = nums[i];
        }
        nums[i] = pivot;
        if (i == k - 1) {
            return nums[i];
        } else if (i > k-1) {
            high = i - 1;
        } else {
            lo = i + 1;
        }
    }

    return nums[k - 1];
}