题目描述:
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
思路:
暴力方法就是:先排序,再取前k个最小的值。
由于是n个整数,我们完全可以使用Java API中的函数完成这一功能,首先把数组中的元素添加到list集合对象中,然后调用Collections.sort()函数,完成排序(这个过程的时间复杂度是O(nlogn),因为该函数的实现采用了快速排序算法),然后把list集合中除前k个数删掉,这样就返回了最小的k个数。
import java.util.ArrayList;
import java.util.Collections;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
ArrayList<Integer> result = new ArrayList<Integer>();
if(input==null||k<=0||input.length==0||input.length<k) return result;
for(int i=0;i<input.length;i++){
list.add(input[i]);
}
Collections.sort(list);
for(int i = 0;i<k;i++){
result.add(list.get(i));
}
return result;
}
}
另一种方法:
创建一个用于保存最小k个数的容器,第一步是放入k个元素,第二步是比较容器中最大的数与当前数组遍历的那个数的大小,如果容器中最大的数比当前遍历的数大,就移除该数,并放入当前遍历的那个数。这样整个循环结束的时候,容器中就是最小的k个数了。
代码实现:
import java.util.ArrayList;
import java.util.Collections;
import java.util.TreeSet;
import java.util.Iterator;
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
if(input == null || k <= 0 || input.length == 0 || k > input.length) return list;
//创建一TreeSet集合,因为TreeSet默认就是从小到大排序的
TreeSet<Integer> s = new TreeSet<Integer>();
for (int i = 0; i < input.length; i++) {
if(s.size() < k){
s.add(input[i]);
}else{
if(input[i] < s.last()){
s.remove(s.last());
s.add(input[i]);
}
}
}
//循环结束,s中已经保存了前k个最小的数,只需要迭代放入ArrayList中即可
Iterator<Integer> it = s.iterator();
while(it.hasNext()){
list.add(it.next());
}
return list;
}
}