题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路分析:
- 先考虑边界情况:k为0时,数组长度n小于k时,数组为空时
- 把前k各元素加入到结果集中(用result表示)
- 分析数组中k+1至n各元素:如果该元素比result中最大值大,则替换
- 返回result
public ArrayList<Integer> getLeastNum(int[] input,int k){
ArrayList<Integer> result=new ArrayList<Integer>();
//为空时
if(input==null){
return null;
}
int len=input.length;
//返回长度大于数组总长度时 以及返回个数为0个时
if(k>len || k==0){
return result;
}
if(k==len){//返回长度等于数组总长度时
for(int i=0;i<len;i++)
result.add(input[i]);
return result;
}else{
for(int i=0;i<k;i++){//先把前k各元素排序加入
result.add(input[i]);
}
for(int j=k;j<len;j++){//循环k--len个元素
int temp=input[j];
//获取已加入的k个元素中最大的元素位置
int max=0;
for(int m=1;m<k;m++){
if(result.get(m)>result.get(max))
max=m;
}
//最大元素与目标元素比较
if(temp<result.get(max)){
result.set(max, temp);
}
}
}
return result;
}