lintcode 第k大元素

描述

在数组中找到第 k 大的元素。

样例

样例 1：

输入：
n = 1, nums = [1,3,4,2]
输出：
4
样例 2：

输入：
n = 3, nums = [9,3,2,4,8]
输出：
4

挑战

要求时间复杂度为O(n)，空间复杂度为O(1)。

思路

使用快速排序的思路，第k大的元素就是排序完成之后的第nums.size() - k位置上的元素。
即qSort(nums, 0, nums.size(), nums.size() - n)
根据基准数调换完位置之后，判断k的位置和i j的关系，进行下一轮的调换。

代码

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

            
        return temp;
        
    }
    
};