剑指offer——最小的K个数__(待优化)

题目描述

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

思路：TreeSet是有序的，全部放进去，然后取出k个来。

题目注意，如果k大于数组长度 ，是要return 空的集合的。

import java.util.ArrayList;import java.util.Iterator;import java.util.TreeSet;public class Solution {    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {         ArrayList<Integer>list = new ArrayList<>();if(input.length==0||k==0||k>input.length)return list;TreeSet<Integer>set = new TreeSet<>();for (int i = 0; i < input.length; i++) {set.add(input[i]);}Iterator<Integer>iterator =set.iterator();int num=0;while(num<k&&iterator.hasNext()){list.add(iterator.next());num++;}return list;    }}