Leetcode--Java--剑指 Offer 40. 最小的k个数

最新推荐文章于 2026-01-06 23:11:15 发布

原创最新推荐文章于 2026-01-06 23:11:15 发布 · 523 阅读

0 ·

本内容遵循CC 4.0 BY-SA版权协议

本文介绍了如何使用快速选择排序算法和大顶堆数据结构解决查找整数数组中最小k个数的问题，通过实例和代码展示了如何在O(n)时间内找到并返回结果。

题目描述

输入整数数组 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);
    }
}