剑指offer :最小的K个数

题目:

//输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。

思路:
采用优先级队列 默认大顶堆 其中元素个数超过k时,每次加入一个元素就删除一个头元素
时间复杂度为O(nlog(k+1))

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) 
    {
        if (k>input.size())
            return vector<int>();
        vector<int> tmp;
        priority_queue<int> que(input.begin(), input.begin()+k);
        for (int i=k;i<input.size();i++)
        {
            que.push(input[i]);
            que.pop();
        }
        while(!que.empty())
        {
            tmp.push_back(que.top());
            que.pop();
        }
        return vector<int> (tmp.rbegin(), tmp.rend());
    }
};

猜你喜欢

转载自blog.csdn.net/pynash123/article/details/89104799