寻找第K大的数
基础知识
构造STL中的最大堆、最小堆,代码如下:
std::priority_queue<int> big_heap; //默认构造是最大堆
std::priority_queue<int, std::vector<int>, //最小堆构造法
std::greater<int> > small_heap;
std::priority_queue<int, std::vector<int>, //最大堆构造法
std::less<int> > big_heap2;
最大堆排列形式如下:
题目
已知一个未排序的数组,求这个数组中第K大的数字。
如,array = [3,2,1,5,6,4] ,k = 2,return 5
思路
这是一道十分容易看懂并动手写出的题,比如我直接把给我的数组进行排序之后,找到相对应下标的数字即可,但我这里将提供一个新的思维,利用最小堆的形式来快速找到这个数字。
比如,我把较大的k个数都放入到最小堆里面,那么最后栈顶的元素就是第K大的数字,这里面分为如下情况:
①当最小堆中的数字个数小于k时,直接放入到最小堆中。
②当最小堆中的数字个数大于等于k时,再比较堆顶的元素是否小于要放入的数,若小于,堆顶元素弹出,将元素放入堆中,若不小于,则不放入。(即找到较大的K个数)
实现代码如下:
#include <stdio.h>
#include <vector>
#include <queue>
class Solution {
public:
int findKthLargest(std::vector<int>& nums, int k) {
std::priority_queue<int, std::vector<int>, std::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(){
std::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 solve;
printf("%d\n", solve.findKthLargest(nums, 2));
return 0;
致谢
本章知识点和思路由小象学院相关视频提供,由本人学习并梳理得出,希望自己加深记忆的同时,也能给大家提供更多有关于一些算法的知识点。
你的点赞、评论、收藏就是对我最大的支持与鼓励,谢谢!