题目:输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
解题思路:先创建一个大小为k的数据容器来存储最小的k个数字,接下来每次从输入的n个整数中读入一个数,如果容器中已有的数字少于k个,则直接从把这次读入的整数放入容器中,如果容器中已经有k个数字了,也就是容器已满,此时我们不能在插入新的数字,而是只能替换已有的数字,找出这已有的k个数中的最大值,然后拿这次待插入的整数和最大值进行比较,如果带插入的值比当前已有的最大值小,则用这个整数替换当前已有最大值,否则,抛弃这个整数。
import java.util.ArrayList;
import java.util.Collections;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> retList = new ArrayList<>();
if(input==null||input.length<=0||input.length<k||k<=0){
return retList;
}
for(int i=0;i<input.length;i++){
if(retList.size()<k){
retList.add(input[i]);
}
if(retList.size()==k){
int detIndex = getMax(retList);
if(input[i]<retList.get(detIndex)){
retList.remove(detIndex);
retList.add(input[i]);
}
}
}
Collections.sort(retList);
return retList;
}
private int getMax(ArrayList<Integer> retList) {
// TODO Auto-generated method stub
int maxIndex = 0;
for(int i=0;i<retList.size();i++){
if(retList.get(i)>retList.get(maxIndex)){
maxIndex = i;
}
}
return maxIndex;
}
}
注:这里需要注意边界条件的判定:
if(input==null||input.length<=0||input.length<k||k<=0)