【LeetCode】215. Kth Largest Element in an Array (Medium)

最新推荐文章于 2021-05-04 23:45:46 发布

liu_shm

最新推荐文章于 2021-05-04 23:45:46 发布

阅读量199

点赞数

分类专栏： leetcode

本文链接：https://blog.csdn.net/liu_shm/article/details/52537622

版权

leetcode 专栏收录该内容

13 篇文章 0 订阅

订阅专栏

【题意】找出第k大的元素

【解】类似快排

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
        return f(nums, 0, nums.size(), k);
    }
private:
    int f(vector<int>& nums, int l, int r, int k) {
	int pivot = nums[l];
	int i = l + 1;
	for (int j = i; j < r; j++) {
		if ((nums[j] < pivot)) {
			int temp = nums[i];
			nums[i] = nums[j];
			nums[j] = temp;
			i++;
		}
	}
	nums[l] = nums[i - 1];
	nums[i - 1] = pivot;
	if (l >= r) return nums[l];
	if (i - l + k == r - l + 1)
		return nums[i - 1];
	else if (i - l + k > r - l + 1)
		return f(nums, l, i - 1, k - r + i - 1);
	else
		return f(nums, i, r, k);
}
};