《剑指offer》面试题30：最小的k个数（java实现）

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

来源于——《剑指offer》

• 思路1：利用快速排序的patition函数， 先随机找到一个中间数字，返回索引值，通过比较索引值和k的大小来判断是否再次调用patition函数，知道k和返回的索引相同。
• 缺点：修改了原来的数组

public class Solution {
    private void swap(int[] array, int i, int j) {
        int temp = array[i];
        array[i] = array[j];
        array[j] = temp;
    }

    private int partition(int[] array, int start, int end) {
        int mid = (start + end) >> 1;
        if (array[start] > array[end]) {
            swap(array, start, end);
        }
        if (array[mid] > array[end]) {
            swap(array, mid, end);
        }
        if (array[mid] > array[start]) {
            swap(array, mid, start);
        }
        int key = array[start];
        while (start < end) {
            while (start < end && key <= array[end]) {
                --end;
            }
            array[start] = array[end];
            while (start < end && key >= array[start]) {
                ++start;
            }
            array[end] = array[start];
        }
        array[start] = key;
        return start;
    }

    public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        ArrayList<Integer> resList = new ArrayList<>();
        if (input == null || input.length == 0 || k <= 0 || k > input.length) {
            return resList;
        }
        int start = 0;
        int end = input.length - 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++) {
            resList.add(input[i]);
        }
        return resList;
    }
}

• 思路2：利用红黑树，先将前k个数装入set中，然后用set中最大值与后面的数比较，如果找到比最大数小的数就将最大数移除，装入当前数，直到数组中的数遍历完毕。

public class Solution2 {
    public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
        ArrayList<Integer> resList = new ArrayList<>();
        if (input == null || input.length == 0 || k <= 0 || k > input.length) {
            return resList;
        }
        TreeSet<Integer> set = new TreeSet<>();
        for (int i = 0; i < input.length; i++) {
            if (i < k) {
                set.add(input[i]);
            } else {
                if (input[i] < set.last()) {
                    set.remove(set.pollLast());
                    set.add(input[i]);
                }
            }
        }
        for (Integer integer : set) {
            resList.add(integer);
        }
        return resList;
    }
}