LeetCode117|最小的k个数

2020-10-27 18:16:56 浏览数 (1)

1,问题简述

输入整数数组 arr ,找出其中最小的 k 个数。

例如,输入4、5、1、6、2、7、3、8这8个数字,则最小的4个数字是1、2、3、4。

2,示例

代码语言:javascript复制
示例 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

3,题解思路

使用排序,队列的方式进行解决

4,题解程序

代码语言:javascript复制

import java.util.Arrays;
import java.util.Collections;
import java.util.PriorityQueue;
import java.util.Queue;

public class GetLeastNumbersTest {
    public static void main(String[] args) {
        int[] arr = {4, 5, 1, 6, 2, 7, 3, 8};
        int k = 4;
        int[] leastNumbers = getLeastNumbers2(arr, k);
        for (int num : leastNumbers
        ) {
            System.out.print(num   "t");
        }
    }

    public static int[] getLeastNumbers(int[] arr, int k) {
        if (arr == null || k == 0) {
            return new int[0];
        }
        Queue<Integer> queue = new PriorityQueue<>((o1, o2) -> o2 - o1);
        for (int num : arr) {
            if (queue.size() < k) {
                queue.offer(num);
            } else if (queue.peek() > num) {
                queue.poll();
                queue.offer(num);
            }
        }
        int[] result = new int[k];

        for (int i = 0; i < k; i  ) {
            result[i] = queue.poll();
        }
        return result;

    }

    public static int[] getLeastNumbers2(int[] arr, int k) {
        if (arr == null || arr.length < k) {
            return new int[0];
        }
        Arrays.sort(arr);
        int[] result = new int[k];
        if (k >= 0) {
            System.arraycopy(arr, 0, result, 0, k);
        }
        return result;
    }
}

5,题解程序图片版

6,总结一下

这道题给出了两个题解思路,两种方式都是比较容易理解的,这里就没有给出过多的注释说明,队列的特点在之前的文章中多次说明了,就是先进先出,但是对于应用的开发者来说,熟悉队列的特点也是有必要的,这里就简单的说下

0 人点赞