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

题意:让你构造一个类,初始化后,每次调用某成员函数可以得到历史数据的第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()).

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
平台声明:文章内容(如有图片或视频亦包括在内)由作者上传并发布,文章内容仅代表作者本人观点,简书系信息发布平台,仅提供信息存储服务。

推荐阅读更多精彩内容

  • 用两张图告诉你,为什么你的 App 会卡顿? - Android - 掘金 Cover 有什么料? 从这篇文章中你...
    hw1212阅读 13,134评论 2 59
  • 最近做项目的时候需要指定特定手机才能使用app,此时必须要根据每台手机特有的IMEI值进行判断。简单说一下IMEI...
    仁昌居士阅读 743评论 0 0
  • 《奔跑吧》数季走过,从换人风波到改名浪潮,观众可以说是对他喜怒参杂。毕竟他陪伴了大家几个盛夏,不管是他的忠实观众还...
    青猕cyan阅读 245评论 0 0