LeetCode 剑指offer 40.最小的k个数

最新推荐文章于 2024-08-09 18:00:00 发布

icgogogo

最新推荐文章于 2024-08-09 18:00:00 发布

阅读量104

点赞数

分类专栏： leetcode

本文链接：https://blog.youkuaiyun.com/javagogogo2850/article/details/117991321

版权

leetcode 专栏收录该内容

10 篇文章

订阅专栏

本文介绍了一种寻找整数数组中最小的K个数的方法，包括直接排序和使用优先队列两种实现方式。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

题目描述

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

思路

一上来就些思路排序取出前k个放在新数组里返回。

代码

import static java.util.Arrays.sort;
class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        sort(arr);
        int[] ret = new int[k];
        for (int i = 0; i <k ; i++) {
            ret[i] = arr[i];

        }
        return ret;
    }
}

题解

方法一排序

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;
    }
}
作者：LeetCode-Solution

思路基本一致

方法二堆，优先队列

class Solution {
    public int[] getLeastNumbers(int[] arr, int k) {
        int[] vec = new int[k];//用于返回的数组
        if (k == 0) { // 排除 0 的情况
            return vec;
        }
        PriorityQueue<Integer> queue = new PriorityQueue<Integer>(new Comparator<Integer>() {//？重写conpare方法，规定比较标准呢
            public int compare(Integer num1, Integer num2) {
                return num2 - num1;//优先队列的比较标准:
            }
        });
        for (int i = 0; i < k; ++i) {
            queue.offer(arr[i]);//数组的前k个元素加入优先队列
        }
        for (int i = k; i < arr.length; ++i) {//让优先队列里的元素为最小的k个
            if (queue.peek() > arr[i]) {//peek查，如果队列元素比数组里元素大
                queue.poll();//poll取，删，取出队列元素
                queue.offer(arr[i]);//加入数组元素
            }
        }
        for (int i = 0; i < k; ++i) {
            vec[i] = queue.poll();
        }
        return vec;
    }
}

作者：LeetCode-Solution

反思总结

1 java.util.Arrays.sort; 方法 sort（数组）；默认升序排列

2 学习Java的优先队列操作

3 怎么把最小的k个加入队列没看懂，后面在琢磨一下