时间效率:最小的K个数

时间:2024-07-21 17:33:02

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

import java.util.ArrayList;
import java.util.Iterator;
import java.util.TreeSet;
/*
* 利用TreeSet排序并去除重复元素,利用ArrayList存储并输出
*/
public class Solution {
public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
ArrayList<Integer> list=new ArrayList<Integer>();
ArrayList<Integer> list2=new ArrayList<Integer>();
if(input==null||input.length==0||k==0||k>input.length)
return list;
TreeSet<Integer> set=new TreeSet<Integer>();
for(int i=0;i<input.length;i++){
set.add(input[i]);
}
Iterator<Integer> it = set.iterator();
while (it.hasNext()) {
int x=it.next();
list.add(x);
}
for(int i=0;i<k;i++){
list2.add(list.get(i));
}
return list2;
}
}

相关文章