面試題30、輸入n個(gè)整數(shù)，找出其中最小的k個(gè)數(shù)，例如輸入4,5,1,6,2,7,3,8這個(gè)8個(gè)數(shù)字，則最小的4個(gè)數(shù)字是1,2,3,4
import java.util.ArrayList;
import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> result = new ArrayList<Integer>();
if(k > input.length || k<=0){
return result;
}
//定義排列的順序
Comparator<Integer> cmp;
cmp = new Comparator<Integer>(){
public int compare(Integer o1, Integer o2){
return o2-o1;
}
};
//java中的最大堆結(jié)構(gòu)
PriorityQueue<Integer> qi = new PriorityQueue<Integer>(k,cmp);
//添加k個(gè)元素,按照從小到大的順序進(jìn)行排列，能夠逆序
int i = 0;
for(;i<k;i++){
qi.add(input[i]);
}
// System.out.println("queue輸出"+qi.size());
// while(!qi.isEmpty()){
// System.out.print(qi.poll()+" ");
// }
for(;i<input.length ;i++){
if(input[i] < qi.peek()){
qi.poll();
qi.add(input[i]);
}
}
while(!qi.isEmpty()){
result.add(qi.poll());
}
return result;
}
}