Lintcode 544 前K大数

使用优先队列的O(Nlogk)解法

class Solution {
public:
    /**
     * @param nums: an integer array
     * @param k: An integer
     * @return: the top k largest numbers in array
     */
    class minOper 
    {
        public:
            bool operator() (const int &a ,const int &b)
            {
                return a > b;
            }
    };
    vector<int> topk(vector<int> &nums, int k) 
    {
        int n = nums.size();
        int i,j;
        vector<int> answer;
        priority_queue<int,vector<int>,minOper> minHeap;
        
        for(i=0;i<k;i++)
        {
            minHeap.push(nums[i]);
        }
        for(i=k;i<n;i++)
        {
            if(nums[i]<=minHeap.top()){
                continue;
            }
            else
            {
                minHeap.pop();
                minHeap.push(nums[i]);                
            }
        }
        while(!minHeap.empty())
        {
            answer.push_back(minHeap.top());
            minHeap.pop();
        }
        reverse(answer.begin(), answer.end());
        return answer;
        // write your code here
    }
};

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值