leetcode215.数组中的第K个最大元素

题目大意

在未排序的数组中找到第 k 个最大的元素。请注意，你需要找的是数组排序后的第 k 个最大的元素，而不是第 k 个不同的元素。

示例 1:

输入: [3,2,1,5,6,4] 和 k = 2
输出: 5

示例 2:

输入: [3,2,3,1,2,4,5,5,6] 和 k = 4
输出: 4

解题思路

方法一：

对数组排序，然后找到第K个数即可；

方法二：

维护一个大小为K个小根堆。若当前元素大于堆顶元素，则弹出堆顶元素，并将当前元素放入堆中；

class Solution {
public:
	int findKthLargest(vector<int>& nums, int k) {
		if (nums.size() == 0 || nums.size() < k)
			return{};
		priority_queue<int, vector<int>, greater<int>>q;
		for (int i = 0; i < k; i++)
		{
			q.push(nums[i]);
		}
		for (int i = k; i < nums.size(); i++)
		{
			if (nums[i] > q.top())
			{
				q.pop();
				q.push(nums[i]);
			}
		}
		return q.top();
	}
};

方法三：

使用快排中的思想。假设当前查找的范围是left~right，寻找第K大的数字。则在范围left ~right上对数组进行一次partition，返回的下标即为index。

• 若right-index==K-1：表示右边大于nums[index]的数字共有K-1个，而我们要寻找第K大的数字，必然是nums[index]；
• 若right-index>K-1：表明大于nums[index]的数字要多于K-1个，因此应该在index ~right中找第K大的数字；
• 否则：表明大于nums[index]的数字不够，因此需要从左部分寻找。此时不是找第K大的元素，而是应该扣除右半部分的元素数量；

class Solution {
public:
    int findKthLargest(vector<int>& nums, int k) {
    	if(nums.size() < k)
    		return -1;
    	return findTopK(nums, k, 0, nums.size() - 1);
    }

    int findTopK(vector<int> & nums, int k, int left ,int right){
    	int index = partition(nums, left, right);
    	if (right - index == k - 1)
    		return nums[index];
    	else if (right - index > k - 1)
    		return findTopK(nums, k, index + 1, right);
    	else
    		return findTopK(nums, k - (right - index) - 1, left, index - 1);
    }

    int partition(vector<int> & nums, int left, int right){
    	int firstNum = nums[left];
    	int head = left + 1, end = right;
    	while (head <= end){
    		while (end >= head && nums[end] >= firstNum)
    			--end;
    		while (end >= head && nums[head] < firstNum)
    			++head;
    		if (end > head)
    			swap(nums[end], nums[head]);
    		else
    			break;
    	}
    	swap(nums[left], nums[end]);
    	return end;
    }
};