美文网首页LeetCode蹂躏集
2018-09-11 703. Kth Largest Elem

2018-09-11 703. Kth Largest Elem

作者: alexsssu | 来源:发表于2018-09-11 18:33 被阅读0次

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

    相关文章

      网友评论

        本文标题:2018-09-11 703. Kth Largest Elem

        本文链接:https://www.haomeiwen.com/subject/qmzugftx.html