5.数组中第K大的数(LeetCode 215)

在未排序的数组中找到第 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时，新元素直接加入堆中 
当堆中元素有k个时，就判断堆顶元素是否小于新元素，如果小于，则将新元素加入堆中

#include<iostream>
#include<queue>
#include<vector>
using namespace std;

class Solution{
    
    
public:
	int findKthLargest(vector<int>& nums, int k) {
    
     
		priority_queue<int,vector<int>,greater<int>> Q;	
		for(int i = 0; i < nums.size(); i++) {
    
    
			if(Q.size() < k) {
    
    
				Q.push(nums[i]);
			}
			else if(Q.top() < nums[i]) {
    
    
				Q.pop();
				Q.push(nums[i];
			}
		}
		return Q.top();	//返回堆顶
	}
};

int main(void) {
    
    
	vector<int> nums;
	nums.push_back(3);
	nums.push_back(2);
	nums.push_back(1);
	nums.push_back(5);
	nums.push_back(6);
	nums.push_back(4);
	
	Solution s;
	int res = s.findKthLargest(nums,2);
	cout<<res; 
	return 0;
}