题目描述:输入n个整数,找出其中最小的K个数
例如输入4,5,6,1,2,3,7,8这8个数字
则最小的4个数字是1,2,3,4
方法一:对数组进行排序 时间复杂度O(nlogn)
import java.util.ArrayList;
import java.util.Arrays;
public class GetLeastNumbers {
public static ArrayList<Integer> getLeastNumbers(int[] input,int k){
ArrayList<Integer> list=new ArrayList<Integer>();
if(input.length==0 ||input==null||k>input.length){
return list;
}
Arrays.sort(input);
if(k==1){
System.out.println(input[0]);
}
for(int i=0;i<k;i++){
list.add(input[i]);
}
return list;
}
public static void main(String[] args) {
System.out.println(getLeastNumbers(new int[]{4,5,6,1,2,3,7,8},4));
}
}
其中Arrays.sort()也可以换为各种排序方法:例如下面的冒泡排序法:
for(int i=0;i<input.length;i++) {
for (int j = i + 1; j < input.length; j++) {
if (input[i] > input[j]) {
int temp = input[i];
input[i] = input[j];
input[j] = temp;
}
}
}
方法二:将数组分成前k个数和后n-k个数,定义一个指针再后部分移动,找出前面部分数中最大数的下标,如果指针指向的数比前k个数的最大值小时,就将这个最大值替换为指针指向的数
public class GetLeastNumbers {
public static ArrayList<Integer> getLeastNumbers(int[] input,int k) {
ArrayList<Integer> list = new ArrayList<Integer>();
if (input.length == 0||input==null||k>input.length){
return list;
}
for(int i=0;i<k;i++){
list.add(input[i]);
}
for(int i=k;i<input.length;i++){
int index=getMaxIndex(list);
if(input[i]<list.get(index)){
list.set(index,input[i]);
}
}
return list;
}
public static int getMaxIndex(ArrayList<Integer> list){
int index=0;
int a=list.get(0);
for(int i=1;i<list.size();i++){
if(list.get(i)>a){
a=list.get(i);
index=i;
}
}
return index;
}
public static void main(String[] args) {
int[] a={4,5,6,1,2,3,7,8};
int k=4;
ArrayList<Integer> list=new GetLeastNumbers().getLeastNumbers(a,k);
for(int i:list){
System.out.print(i+" ");
}
}
}