题解 | #最小的K个数#
最小的K个数
https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf
import java.util.*; public class Solution { /** * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可 * * * @param input int整型一维数组 * @param k int整型 * @return int整型ArrayList */ public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) { // write code here ArrayList<Integer> res = new ArrayList<Integer>(); //建立大根堆,建立优先队列使用lambda表达式 初始化方式 为后者-前者 PriorityQueue<Integer> p = new PriorityQueue<Integer>((p1, p2) ->p2 - p1); if(k==0||input.length==0){ return res; } //先存满k个元素 for (int i = 0; i < input.length; i++) { if (p.size() < k) { p.offer(input[i]); } else { if (input[i] < p.peek()) { p.poll(); p.offer(input[i]); } } } for(int num:p){ res.add(num); } // Iterator<Integer> iterator = p.iterator(); // while (iterator.hasNext()) { // res.add(iterator.next()); // } Collections.sort(res); return res; } }