标签：util rev code 剑指offer 顺序 java integer int 处理
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。
最先想到的是将数组升序排列，返回前k个元素。不过排序的话效率可能会较低，我们可以使用优先级队列模拟堆来处理。
模拟一个k个元素的最大堆，当堆内元素个数等于k的时候，新来的元素就要和堆顶元素去比较，如果小于堆顶元素，就删除堆顶元素，将新元素添加进堆中，由于最大堆的堆顶元素是最大元素，我们可以保持堆中元素始终是最小的几个，因为每有新的元素来，都会进行比较，最后堆中的元素便是前k个最小元素了。
吐个槽，这种题最后输出又不需要顺序了，牛客网判题真的迷。
C++
class Solution { public: vector<int> GetLeastNumbers_Solution(vector<int> input, int k) { if(input.size() == 0 || k == 0 || k > input.size()) return res; for(int num:input){ if(q.size() < k) q.push(num); else{ if(num < q.top()){ q.pop(); q.push(num); } } } while(!q.empty()){ res.push_back(q.top()); q.pop(); } //reverse(res.begin(), res.end()); return res; } private: vector<int> res; priority_queue<int> q; };
Java
import java.util.*; public class Solution { public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) { if(input.length == 0 || k == 0 || k > input.length) return res; for(int num:input){ if(queue.size() < k) queue.add(num); else{ if(num < queue.peek()){ queue.poll(); queue.add(num); } } } for(Integer i : queue) { res.add(i); } return res; } static Comparator<Integer> cmp = new Comparator<Integer>() { public int compare(Integer e1, Integer e2) { return e2 - e1; } }; private ArrayList<Integer> res = new ArrayList<>(); private PriorityQueue<Integer> queue = new PriorityQueue<>(cmp); }
标签：util rev code 剑指offer 顺序 java integer int 处理
原文地址：https://www.cnblogs.com/silentteller/p/11959368.html
