最小的K个数

题目描述
输入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) {
        if (input == null) return null;
        ArrayList<Integer> ans = new ArrayList<>();
        if (k>input.length || k == 0) return ans;
        PriorityQueue<Integer> pq = new PriorityQueue<Integer>(new Comparator<Integer>(){
			@Override
			public int compare(Integer o1, Integer o2) {
                return o2.compareTo(o1);
            }
		});
        for (int i = 0; i<input.length; i++){
            if (pq.size()<k) pq.offer(input[i]);
            else if (input[i] < pq.peek()){
                pq.poll();
                pq.offer(input[i]);
            }
        }
        
        while (!pq.isEmpty()){
            ans.add(pq.poll());
        }
        return ans;
    }
}