Leetcode 215.Kth Largest Element in an Array

Leetcode

215. Kth Largest Element in an Array

最naive的做法自然就是all in sort

sort(nums.begin(),nums.end());
return nums[nums.size() - k];

time complexity: O(nlogn)

然而我们只需要找到第k个最大的数 这样有太多的浪费的操作了

可以使用quicksort，quicksort每次可以把一个pivot放到正确的位置，于是我们就可以根据pivot的index与k比较继续用左边或者右边找下一个pivot，直至找到位置为k的pivot,这里给一个非递归的
time complexity：O(n)
worst: O(n^2)

int len = nums.size();
int i = 0;
int j = len - 1;
while (true) {
    int temp = nums[i];
    while (i < j) {
        while (nums[j] >= temp && i < j) {
            j--;
        }
        nums[i] = nums[j];
        while (nums[i] <= temp && i < j) {
            i++;
        }
        nums[j] = nums[i];
    }
    nums[i] = temp;
    if (len - i < k) {
        j--;
        i = 0;
    } else if (len - i > k) {
        i++;
        j = len - 1;
    } else {
        return nums[i];
    }
}

也可以用heap sort来找第k个最大的数
下面用priority queue实现

priority_queue<int> q(nums.begin(),nums.end());
for (int i = 1; i < k; i++) q.pop();
return q.top();