题目
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
思路
- 先判断k的值是不是越界的情况!!最开始本人就是因为这个错了。
- 先将数组排序。这里可用提供的sort函数
- 再将前k个数存入到ArrayList即可。
AC代码
import java.util.ArrayList;
import java.util.Arrays;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
if(k > input.length || k <= 0)
return list;
Arrays.sort(input);
int i = 0;
while(k-- > 0) {
list.add(input[i++]);
}
return list;
}
}