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();