题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
PriorityQueue<Integer> q=new PriorityQueue<Integer>();
ArrayList<Integer> ans=new ArrayList<Integer>();
if(input==null||k>input.length) return ans;
for(int i=0;i<input.length;i++){
q.offer(input[i]);
}
while(!q.isEmpty()&&k>0){
ans.add(q.poll());
k--;
}
return ans;
}
}