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

电玩女神 2022-11-15 11:17 314阅读 0赞

解题思路：

（1）使用优先队列，priority_queue

  1. class Solution {
  2. public:
  3.     vector<int> GetLeastNumbers_Solution(vector<int> input, int k) {
  4.         vector<int> v;
  5.         priority_queue<int,vector<int>,less<int>> pq;
  6.         if(k>input.size() || k==0) return v;
  8.         for(int i=0;i<input.size();i++) {
  9.             if(pq.empty()) pq.push(input[i]);
  10.             else if(pq.size()<k) pq.push(input[i]);
  11.             else {
  12.                 pq.push(input[i]);
  13.                 pq.pop();
  14.             }
  15.         }
  17.         while(!pq.empty()) {
  18.             v.push_back(pq.top());
  19.             pq.pop();
  20.         }
  21.         return v;
  22.     }
  23. };

发表评论

表情：
评论列表 （有 0 条评论，314人围观）

还没有评论，来说两句吧...

相关阅读