BM46 最小的K个数

1.题目描述

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围：0≤k,n≤100000≤k,n≤10000，数组中每个数的大小0≤val≤10000≤val≤1000

要求：空间复杂度 O(n)O(n) ，时间复杂度 O(nlogk)O(nlogk)

示例1

输入：

[4,5,1,6,2,7,3,8],4 

返回值：

[1,2,3,4]

说明：

返回最小的4个数即可，返回[1,3,2,4]也可以        

示例2

输入：

[1],0

返回值：

[] 

示例3

输入：

[0,1,2,1,2],3

返回值：

[0,1,1]

2.解题思路

使用小根堆可以完美解决本题，对于input数组中的每一个整数，将其加入到小根堆中，然后依次弹出堆中的前k个元素即为所求的最小的k个数

3.代码实现

import java.util.*;   public class Solution {     /**      * 代码中的类名、方法名、参数名已经指定，请勿修改，直接返回方法规定的值即可      *      *      * @param input int整型一维数组      * @param k int整型      * @return int整型ArrayList      */     Comparator<Integer> comparator = new Comparator<Integer>() {         @Override         public int compare(Integer i, Integer j) {             if (i > j) return 1;             else if (i < j) return -1;             else return 0;         }     };      public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {         // write code here         ArrayList<Integer> res = new ArrayList<>();         PriorityQueue<Integer> pq = new PriorityQueue<>(comparator);         for (int i = 0; i < input.length; i++) {             pq.offer(input[i]);         }         for (int i = 0; i < k; i++) {             if (!pq.isEmpty()) {                 res.add(pq.poll());             }         }         return res;     } }