剑指offer最小的k个数

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

思路:
最直观的思路就是采用一个长度为k的数组nums,保证数组里边的数有序,每访问输入数组input的一个节点,将其与nums数组的值最大的元素即nums[nums.length - 1]进行比较,如果input节点更小,则替换。
本题也要注意各种异常的排除,包括:
输入数组为空,输入的k值为0,k值大于数组长度。

参考代码:


import java.util.ArrayList;
import java.util.Arrays;

public class MinKNumbers {
	public static ArrayList GetLeastNumbers_Solution(int[] input, int k) {
		if (input == null || input.length < k || k == 0) {
			return new ArrayList();
		}
		int[] num = new int[k];
		for (int p = 0; p < k; p++) {
			num[p] = input[p];
		}
		Arrays.sort(num);
		System.out.println(Arrays.toString(num));
		for (int i = k - 1; i < input.length; i++) {
			if (input[i] < num[k - 1]) {
				num[k - 1] = input[i];
				Arrays.sort(num);
			}
		}
		ArrayList aList = new ArrayList<>();
		for (int j = 0; j < k; j++) {
			aList.add(num[j]);
		}
		return aList;
	}

	public static void main(String[] args) {
		int[] input = { 4, 5, 1, 6, 2, 7, 3, 8 };
		int k = 8;
		ArrayList arrayList = GetLeastNumbers_Solution(input, k);
 		for(int i: arrayList) {
			System.out.print(i+" ");
		}
	}
}

你可能感兴趣的:(算法学习)