最小的K个数

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

import java.util.ArrayList;
import java.util.PriorityQueue;
public class Solution {
    
	public static ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if (k>input.length||k==0) {
        	return new ArrayList<Integer>();
        }
        PriorityQueue<Integer> p = new PriorityQueue<Integer>(k);
        ArrayList<Integer> arrayList = new ArrayList<Integer>();
        for (int i=0; i<input.length; ++i) {
        	p.add(input[i]);
        }
        
	for (int i=0; i<k; ++i) {
		if (!p.isEmpty()) {
			arrayList.add(p.poll());
		}
			
	}
        
	return arrayList;
    }	
}