牛客网——最小的K个数（C++）

输入一个矩阵，按照从外向里以顺时针的顺序依次打印出每一个数字，例如，如果输入如下4 X 4矩阵： 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 则依次打印出数字1,2,3,4,8,12,16,15,14,13,9,5,6,7,11,10.

C++

class Solution {
public:
    vector<int> GetLeastNumbers_Solution(vector<int> input, int k) 
    {
        vector<int> res;
        if(k>input.size())
        {
            return res;
        }
        priority_queue<int> que;
        for(auto it:input)
        {
            que.push(it);
            if(que.size()>k)
            {
                que.pop();
            }
        }
        while(que.size())
        {
            int it=que.top();
            que.pop();
            res.push_back(it);
        }
        return res;
    }
};