C++ 如何利用最小堆找到第K大的数【优先级队列】

寻找第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;

致谢

本章知识点和思路由小象学院相关视频提供，由本人学习并梳理得出，希望自己加深记忆的同时，也能给大家提供更多有关于一些算法的知识点。
你的点赞、评论、收藏就是对我最大的支持与鼓励，谢谢！