一、题目描述
输入n个整数,找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
二、题目分析
- 典型的排序题目,直接使用-快速排序
- 注意一下,input数组的空值和input.length的值与K值的比较
三、题目代码
package offer; import java.util.ArrayList; /* * 最小K个数 */ public class Test28 { static ArrayList<Integer> list = new ArrayList<Integer>(); public static void main(String[] args) { Test28 test28 = new Test28(); int[] input = new int[] { 4, 5, 1, 6, 2, 7, 3, 8 }; list = test28.GetLeastNumbers_Solution(input, 4); System.out.println(list); } public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) { if (input.length == 0 || input.length < k) { return list; } sort(input, 0, input.length - 1); for (int i = 0; i < k; i++) { list.add(input[i]); } return list; } void sort(int[] input, int start, int end) { if (start > end) { return; } int i = start; int j = end; int key = input[i]; while (i < j) { while (i < j && input[j] >= key) { j--; } input[i] = input[j]; while (i < j && input[i] <= key) { i++; } input[j] = input[i]; } input[i] = key; sort(input, start, i - 1); sort(input, i + 1, end); } }