面试题40. 最小的k个数

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

示例 1：

输入：arr = [3,2,1], k = 2
输出：[1,2] 或者 [2,1]

示例 2：

输入：arr = [0,1,2,1], k = 1
输出：[0]

限制：

0 <= k <= arr.length <= 10000
0 <= arr[i] <= 10000

来源：力扣（LeetCode）
链接：https://leetcode-cn.com/problems/zui-xiao-de-kge-shu-lcof
著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。

思路：直接想到的是排序然后取topk.但是这样有点拉。可以使用快排思想优化，只要排到前k个就可以了
解法一：

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if(k == arr.length) {
            return arr;
        }
        partition(arr,0,arr.length - 1,k);
        return Arrays.copyOfRange(arr,0,k);
    }

    public void partition(int [] arr, int start, int end,int k) {
        int left = start;
        int right = end;
        int p = arr[left];
        while(left < right) {
            while(left < right && arr[right] >= p) {
                right--;
            }
            if(left < right) {
                arr[left] = arr[right];
            }
            while(left < right && arr[left] <= p) {
                left++;
            }
            if(left < right) {
                arr[right] = arr[left];
            }
        }
        arr[right] = p;
        if(right == k) {
            return;
        }
        if(right > k) {
            partition(arr,start,right - 1,k);
        }else{
            partition(arr,right + 1,end,k);
        }
    }
}

解法二：维护一个小顶堆，取前k. 或者维护一个大顶堆，让堆中始终最多存放k个值

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        if(k == arr.length) {
            return arr;
        }
        PriorityQueue <Integer> PriorityQueue = new PriorityQueue<>();
        for(int i = 0; i < arr.length; i++) {
            PriorityQueue.add(arr[i]);
        }
        int [] res = new int [k];
        for(int i = 0;i < k;i ++) {
            res[i] = PriorityQueue.poll();
        }
        return res;
    }
}