剑指offer：JZ29 最小的K个数

题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。
示例1
输入
[4,5,1,6,2,7,3,8],4
返回值
[1,2,3,4]
在线链接
代码：

import java.util.*;
public class Solution {

    ArrayList<Integer> list = new ArrayList<>();

    public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        if (input.length < k || k <= 0) {
            return list;
        }
        partition(input, 0, input.length - 1, k - 1);
        return list;
    }

    private void partition(int[] nums, int l, int r, int k) {
        swap(nums, l, (int) (l + (r - l + 1) * Math.random()));
        int less = l - 1;
        int more = r;
        int left = l;
        while (l < more) {
            if (nums[l] < nums[r]) {
                swap(nums, ++less, l++);
            } else if (nums[l] > nums[r]) {
                swap(nums, l, --more);
            } else {
                l++;
            }
        }
        swap(nums, r, more);
        if (l == k) {
            for (int i = 0; i <= l; i++) {
                list.add(nums[i]);
            }
        } else if (l > k) {
            partition(nums, left, l - 1, k);
        } else {
            partition(nums, l + 1, r, k);
        }
    }

    private void swap(int[] nums, int i, int j) {
        int temp = nums[i];
        nums[i] = nums[j];
        nums[j] = temp;
    }
}

思路概述：
这是一个topK问题，只不过不是输出第K大的数，而是输出最小的K个数，思路不变，同样里利用快排的特性。我们知道，每进行一次快排，我们可以确定一个元素（如果有相同元素，那就是这一组相同元素）的位置

比如说，红色的元素是排好的元素，我们可以确定，前面两个元素肯定是最小的两个，后面三个是最大的三个。所以我们利用快排的特性来做这个题。
比如一次遍历后是这样的情况，如果l == k-1，代表已经找到最小的k个数了，直接存储即可
如果l > k-1，代表多了，那我们只需要排前半部分即可（left – l-1)
如果l < k- 1，代表数量不够，那我们只需要排后半部分即可(l + 1 — r)