Leetcode--Java--剑指 Offer 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]

思路

快速选择排序 O(n) + top问题
没必要整个都排序，直接排序找到前topk小的即可。因此可以用快速选择排序，每次根据划分点分段进行查找，本题与这题雷同
复制指定长度数组方法
Arrays.copyOf(originalArray, newLength);
当找到第k大的元素，在第k - 1个位置，就复制它以及它左边的数组元素（排序后的）
方法二：大顶堆

代码

class Solution {

    public int partition(int[] arr, int l, int r, int k) {
        //如果已经缩小到一，就返回位置k
        if (l >= r) return k;
        int i = l - 1, j = r + 1, p = arr[i + j >> 1];
        while (i < j) {
            do i ++; while (arr[i] < p);
            do j --; while (arr[j] > p);
            if (i < j) {
                int t = arr[i];
                arr[i] = arr[j];
                arr[j] = t;
            }
        }
        return j;
    }

    public int[] quickSelect(int[] arr, int l, int r, int k) {
        int j = partition(arr, l, r, k);
        if (j == k) {
            //查找到后，就复制其以及左边部分  copy(arr, newLength)
            return Arrays.copyOf(arr, j + 1);
        }
        //根据位置j来划分，朝某一般进行快排
         return k <= j ? quickSelect(arr, l, j, k) : quickSelect(arr, j + 1, r, k);
    }

    public int[] getLeastNumbers(int[] arr, int k) {
        return quickSelect(arr, 0, arr.length - 1, k - 1);
    }
}