查找数组中的第K个最大元素或者数组中最大的k个元素，C++解法

题目
给定整数数组 nums 和整数 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或者当前遍历的元素大于堆顶元素，那就让当前元素入堆，然后再弹出一个最小的元素。这样就能保证这个堆中保存的时数组中最大的前k个元素。同时堆顶又是堆中最小的一个那么就是数组中第k最大元素。

代码
实现代码使用STL priority_queue来实现最小堆

#include <stdio.h>
#include <vector>
#include <algorithm>
#include <queue>
using namespace std;

class Solution
{
public:
    int findKthLargest(vector<int> &nums, int k)
    {
        int result = 0;
        int numsSize = int(nums.size());
        if (numsSize == 0 || k > numsSize)
        {
            return 0;
        }

        priority_queue<int, vector<int>, greater<int>> store;
        //堆中维持k个最大数
        for (int i = 0; i < numsSize; i++)
        {
            if (int(store.size()) < k || nums[i] > store.top())
            {
                store.push(nums[i]);
                if (int(store.size()) > k)
                {
                    store.pop();
                }
            }
            
            
        }

        result = store.top();
        return result;
    }
};