最小的k个数

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

思路一：利用快排的思想，随机从数组中取一个数X(我就取数组的第一个数了)，如果定义一个游标 i=0 和 j=arr.length-1 ，先从j开始，向前走，如果下标为j的数字<x,那么下标为 j 的数换成X，X的位置换成下表为 j 的数,接着在从前往后走，如果碰到下标为 i 的数>X则将下标为 i 的数换成X，x的位置换成 下标为 i 的数，以此类推，先从后走，再从前走，直到 i=j的时候，此时X左边的数都是小于它的，右边的数都是大于他的，如果X的下标正好是k，那么直接返回k前边的数，否则把数组分为 X 的左边和 X 的右边，分别将两边的数组执行从后向前走，和从前向后走的操作。

代码：

 public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
	 if (k >= input.length)
	 return new ArrayList<>();
	 int[] a = sort(input, 0, input.length - 1, k);
	 for (int i = 0; i < a.length; i++) {
	 System.out.print(a[i]);
	 }
	 List<Integer> list = new ArrayList<Integer>();
	 for (int i = 0; i < k; i++) {
	 list.add(a[i]);
	 }
	 return (ArrayList<Integer>) list;
	 }
	
	 public int partition(int a[], int lo, int hi) {
	 int x = a[lo];
	 int i = lo;
	 int j = hi;
	 while (i < j) {
	 while (x < a[j] && i < j) {
	 j--;
	 }
	 a[i] = a[j];
	 while (x > a[i] && i < j) {
	 i++;
	 }
	 a[j] = a[i];
	 }
	 a[j] = x;
	 return j;
	 }
	
	 public int[] sort(int a[], int lo, int hi, int k) {
	
	 if (lo >= hi) {
	 return a;
	 }
	
	 int index = partition(a, lo, hi);
	 if (index == k - 1) {
	 return a;
	 }
	 sort(a, lo, index - 1, k);
	 sort(a, index + 1, hi, k);
	 return a;
	 }

 

此种方法效率并不高，牛客超时。

思路二：先保存数组前k个数，然后遍历数组第k个以后的数，如果遇到比保存的数中最大的数还要大的，就将其保存起来，并将最大的删除，我们在这我们借用PriorityQueue这个类来保存前k个数。

代码：

	// 方法二：用PriorityQueue类，如果不给他一个比较器，它默认创建的是一个下顶堆，
	public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
		System.out.println(k > input.length || k <= 0);
		if ((k > input.length) || (k <= 0)) {

			return new ArrayList<Integer>();
		}

		PriorityQueue<Integer> queue = new PriorityQueue<Integer>(k, new Comparator<Integer>() {// 给他一个比较器，让他创建一个大顶堆
			@Override
			public int compare(Integer o1, Integer o2) {
				return o2 - o1;
			}
		});
		for (int i = 0; i < input.length; i++) {
			System.out.println("---" + input[i] + "--size=" + queue.size());
			if (queue.size() < k) {
				queue.offer(input[i]);
			} else {
				if (queue.peek() > input[i]) {

					queue.poll();
					queue.offer(input[i]);
				}
			}

		}

		return new ArrayList<Integer>(queue);
	}

此种方法适合处理海量数据