【剑指offer】-最小K个数-28/67

一、题目描述

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

二、题目分析

1. 典型的排序题目，直接使用-快速排序
2. 注意一下，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);
	}

}