java之最小的k个数

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

分析：第一种方法，想到了partition函数，在位置k分割成两个部分，第一部分为小于第k个数，第二部分为大于第k个数，只用知道在k位置之前的数就是我们所求的数字。有个缺点就是改变初始数组，如果要求不改变，可以用第二种解法。

第二种方法，用到了红黑色，就是TreeSet，适合大数据。

java代码：

package LinkList;

import java.util.Iterator;
import java.util.TreeSet;

public class GetLeastNumbersMian {

	// 解法一，用快速排序中的partition函数来解这个问题。但是有个问题是，partition函数会改变数组的元素的顺序
	// 如果面试官要求在不改变输入的数组的情况下，可以用解法二。
	public void GetLeastNumbers1(int[] input, int n, int[] output, int k) {
		if (input == null || output == null || k > n || n <= 0 || k <= 0)
			return;
		int start = 0;
		int end = n - 1;
		int index = Partition(input, start, end);
		while (index != k - 1) {
			if (index > k - 1) {

				end = index - 1;
				index = Partition(input, start, end);

			} else {
				start = index + 1;
				index = Partition(input, start, end);
			}
		}

		for (int i = 0; i < k; i++) {
			System.out.print(input[i] + " ");
			output[i] = input[i];
		}

	}

	// 解法二,使用红黑树来解决，适合海量数据。不会改变初始数组。
	public void GetLeastNumbers(int[] input, TreeSet<Integer> treeset, int k) {
		treeset.clear();
		if (k < 1 || input.length < k)
			return;
		for (int i = 0; i < input.length; i++) {
			if (treeset.size() < k)
				treeset.add(input[i]);
			else {
				int a = treeset.last();
				if (input[i] < a) {
					treeset.remove(a);
					treeset.add(input[i]);
				}

			}

		}
		Iterator<Integer> it = treeset.iterator();
		while (it.hasNext()) {
			System.out.print(it.next() + " ");
		}
	}

	private int Partition(int[] a, int p, int q) {
		int x, i, j, temp;

		x = a[p]; // 将最后一个值保存在x中
		i = p; // 开始的时候将i 移动到数组的外面
		for (j = p + 1; j <= q; j++) {
			if (a[j] <= x) // 比较比x大的数，并交换位置
			{
				i += 1;
				temp = a[i]; // exchange
				a[i] = a[j];
				a[j] = temp;
			}

		}

		temp = a[i]; // exchange
		a[i] = a[p];
		a[p] = temp;

		return i;

	}

	public static void main(String[] args) {
		GetLeastNumbersMian getLeastNumbersMian = new GetLeastNumbersMian();
		int[] input = { 4, 5, 1, 6, 2, 7, 3, 8 };
		int k = 4;
		int output[] = new int[4];
		int n = input.length;
		getLeastNumbersMian.GetLeastNumbers1(input, n, output, k);
		System.out.println();
		TreeSet<Integer> treeset = new TreeSet<Integer>();
		getLeastNumbersMian.GetLeastNumbers(input, treeset, k);

	}

}

最后结果为：

3 1 2 4 
1 2 3 4