题解 | #最小的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,使用优先队列,生成顶最小队列 PriorityQueue<Integer> q = new PriorityQueue<>(); for(int i = 0; i < input.length; i ++){ q.offer(input[i]); } if(q.size() < k) return null; ArrayList<Integer> res = new ArrayList<>(); for(int i = 0; i < k; i ++){ res.add(q.poll()); } return res; } }