剑指offer——最小的K个数
问题描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4。
思路分析
1.先判断K的大小,若K大于输入的数字个数,则返回所有的这些数。
2.对输入的这组数进行排序,此处是从大到小进行排序,然后选取排序后末尾的K个数。
补充:
排序的相关问题,会再写一篇博客说明。
代码
import java.util.ArrayList;
import java.util.Scanner;
class Solution{
public ArrayList<Integer> GetLeastNumbers_Solution(int[] input,int k){
ArrayList<Integer> al = new ArrayList<Integer>();
if (k > input.length) {
return al;
}
for (int i = 0; i < k; i++) {
for (int j = 0; j < input.length - i - 1; j++) {
if (input[j] < input[j + 1]) {
int temp = input[j];
input[j] = input[j + 1];
input[j + 1] = temp;
}
}
al.add(input[input.length - i - 1]);
}
return al;
}
}