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

X-musk

已于 2023-01-22 17:29:24 修改

阅读量50

点赞数 1

分类专栏：剑指Offer(第2版) 文章标签：算法数据结构排序算法 java

于 2023-01-22 17:11:58 首次发布

本文链接：https://blog.csdn.net/WWXDwrn/article/details/128741024

版权

剑指Offer(第2版) 专栏收录该内容

41 篇文章 0 订阅

订阅专栏

该问题要求从给定的整数数组arr中找出最小的k个数。解决方案是先使用Arrays.sort()方法对数组进行升序排序，然后返回排序后的前k个元素。示例展示了当k分别为2和1时的结果。代码实现中定义了一个Solution类，包含一个getLeastNumbers方法，该方法实现了上述逻辑。

摘要由CSDN通过智能技术生成

题目

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

思路：

对原数组从小到大排序后取出前 k 个数即可。

Arrays.sort(int [] a)：对一个数组的所有元素进行排序，且排序时是从小到大。

代码

import java.util.Arrays;

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int[] vec = new int[k];
        Arrays.sort(arr);
        for (int i = 0; i < k; ++i) {
            vec[i] = arr[i];
        }
        return vec;
    }
}