算法与数据结构面试题(5)-查找最小的k 个元素

题目

输入n 个整数，输出其中最小的k 个。
例如输入1，2，3，4，5，6，7 和8 这8 个数字，则最小的4 个数字为1，2，3 和4。

理解

1.其实之前看线性查找算法的时候就一直有一个疑问，像这种查找第K个最小数或者K个最小的数，为什么要用线性查找算法这么难以理解的算法，难道随便用一个排序算法后，再遍历获得第K个小的数或者K个最小的数不好么？

2.既然这一题没有时间上的要求，那就试试看行不行，用快速排序试试。

快速排序

代码

public class Problem5 {
	public int[] sort(int[] datas) {
		if (datas.length == 0 || datas.length == 1)
			return datas;

		List<Integer> left = new ArrayList<Integer>();
		List<Integer> right = new ArrayList<Integer>();
		int pivot = datas[0];
		for (int i = 1; i < datas.length; i++) {
			if (datas[i] <= pivot) {
				left.add(datas[i]);
			} else {
				right.add(datas[i]);
			}
		}
		int[] sorted = new int[datas.length];
		int[] leftArray = new int[left.size()];
		for (int i = 0; i < left.size(); i++) {
			leftArray[i] = left.get(i);
		}
		leftArray = sort(leftArray);
		int[] rightArray = new int[right.size()];
		for (int i = 0; i < right.size(); i++) {
			rightArray[i] = right.get(i);
		}
		rightArray = sort(rightArray);
		int index = 0;
		for (int i = 0; i < leftArray.length; i++) {
			sorted[index++] = leftArray[i];
		}
		sorted[index++] = pivot;
		for (int i = 0; i < rightArray.length; i++) {
			sorted[index++] = rightArray[i];
		}
		return sorted;
	}

	public static void main(String[] args) {
		int[] datas = { 4, 1, 2, 56, 24, 5, 6, 97, 8, 0, 4 };
		int[] sorted = new Problem5().sort(datas);
		print(sorted, sorted.length);
		// 求k个最小的数
		int k = 4;
		print(sorted, k);

	}

	public static void print(int[] datas, int length) {
		for (int i = 0; i < length; i++) {
			System.out.print(datas[i] + " ");
		}
		System.out.println("");
	}
}

输出

0 1 2 4 4 5 6 8 24 56 97 
0 1 2 4