import java.util.*;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();//首先建立一个容器,用来盛放结果
if(input.length == 0|| k == 0 || k > input.length) return list;//特殊情况处理
//遍历输入数组的每一个数
for(int i = 0; i < input.length; i++) {
//当容器中的元素个数小于k个时,直接将数放进容器
if (list.size() < k) {
list.add(input[i]);
}
//如果容器中的个数已经是k个,先对容器进行排序
else {
Collections.sort(list);
//比较当前数值与容器中的最大值(默认升序)比较,如果小于,则删除容器中的最大值,将当前数值放入
if(input[i] < list.get(k-1)) {
list.remove(k-1);
list.add(input[i]);
}
}
}
return list;
}
}
ArrayList类的用法:
ArrayList<Integer> list = new ArrayList<Integer>();
表示生成一个名为list的容器,用来存放int类型。
list.add(a)
添加元素到容器list中;
list.get(i)
获得容器list中索引为i的元素;
list.remove(i)
删除容器list中索引为i的元素,删除后后面的元素会依次向前移动一位。
Collections.sort(list)
对容器list中的元素进行排序(默认升序),需要引入Colletions类(import java.util.Colletions;)