Leetcode 数据流的第K大数值

        设计一个找到数据流中第 k 大元素的类（class）。注意是排序后的第 k 大元素，不是第 k 个不同的元素。

class KthLargest {
private:
    priority_queue<int,vector<int>,greater<int>> q;//实现小根堆
    int k;

public:
    KthLargest(int k, vector<int>& nums) {
        this->k = k;
        for(auto& num:nums){
            add(num);
        }
    }
    
    int add(int val) {
        q.push(val);
        if(q.size() > k){
            q.pop();
        }
        return q.top();

    }
};

本体要注意优先队列实现小根堆的使用：

优先队列有三个参数，其声明形式为：

priority_queue< type, container, function >

这三个参数，后面两个可以省略，第一个不可以。

其中：

1. type：数据类型；
2. container：实现优先队列的底层容器；
3. function：元素之间的比较方式；

对于container，要求必须是数组形式实现的容器，例如vector、deque，而不能使list。

在STL中，默认情况下（不加后面两个参数）是以vector为容器，以 operator< 为比较方式，所以在只使用第一个参数时，优先队列默认是一个最大堆，每次输出的堆顶元素是此时堆中的最大元素。

 

大顶堆（降序）

//构造一个空的优先队列（此优先队列默认为大顶堆）
priority_queue<int> big_heap;   

//另一种构建大顶堆的方法
priority_queue<int,vector<int>,less<int> > big_heap2;   

小顶堆（升序）

//构造一个空的优先队列,此优先队列是一个小顶堆
priority_queue<int,vector<int>,greater<int> > small_heap;   

注意事项

需要注意的是，如果使用less和greater，需要头文件：

#include <functional>