题目
给定整数数组 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;
}
};