2018-09-11 703. Kth Largest Element in a Stream

2024年2月1日 151次阅读 来源: alexssssu

题意：让你构造一个类，初始化后，每次调用某成员函数可以得到历史数据的第K大的数（有两个相同的数算两个）。
解题思路：
思路一：初始化使用vector排序，每次数据来了重新排序，取下标第K个。提交,TLE；
思路二：初始化使用vector排序并取前K个数，每次新数据来了加入vector后再排序，取第K个。提交TLE。
思路三：初始化使用vector排序并取前K个数，每次新数据来了使用二分查找得到新元素应该在的位置，用vector.insert函数插入新数据，取第K个。提交AC，超过25%，很慢。
思路四：成员数据结构使用优先队列，并规定从小到大排序。初始化使用vector排序并取前K个数放入优先队列中，每次新数据来了，将新数据放入优先队列，如果队列长度大于K，剔除最上面的元素，返回优先队列最顶上的元素（最小），提交AC，超过98%。

class KthLargest {
private:
    int kth;
    priority_queue<int, vector<int>, greater<int> > q;
public:
    KthLargest(int k, vector<int> nums) {
        sort(nums.begin(), nums.end(), [](const int a, const int b){return a > b;});
        int sizeN = nums.size();
        for(int i = 0; i < k && i < sizeN; i++)
            q.push(nums[i]);
        kth = k;
    }
    
    int add(int val) {
        q.push(val);
        if(q.size() > kth)
            q.pop();
        return q.top();
    }
};

static int x = [](){
    std::ios::sync_with_stdio(false);
    std::cin.tie(NULL);
    return 0;
}();

/**
 * Your KthLargest object will be instantiated and called as such:
 * KthLargest obj = new KthLargest(k, nums);
 * int param_1 = obj.add(val);
 */

注意：
C++优先队列的使用方法成员函数与普通队列几乎一样；
优先队列使用的其实是堆排序，大顶堆和小顶堆。
priority_queue<int> q1; //大顶堆。声明的时候默认是按照从大到小的顺序（大的在前面，先出队列）
priority_queue<int, vector<int>, greater<int> > q2 ; // 小顶堆。指定队列内装的元素类型（int），队列的容器（vector<int>）, 队列的排序方式（从小到大排序，小的在队列前面，q.pop()和q.top()）.

    原文作者：alexssssu
    原文地址: https://www.jianshu.com/p/73e24477257d
    本文转自网络文章，转载此文章仅为分享知识，如有侵权，请联系博主进行删除。
点赞