LeetCode 面试题 17.14. 最小K个数

最新推荐文章于 2021-09-03 09:28:31 发布

_天道酬勤_不忘初心

最新推荐文章于 2021-09-03 09:28:31 发布

阅读量284

点赞数

分类专栏： LeetCode

本文链接：https://blog.csdn.net/qq_25605637/article/details/105055146

版权

LeetCode 专栏收录该内容

22 篇文章 0 订阅

订阅专栏

思路

题目地址：https://leetcode-cn.com/problems/smallest-k-lcci/
思路：最小堆的简单应用

代码

public class Solution {

    public int[] smallestK(int[] arr, int k) {
        int[] result = new int[k];
        PriorityQueue<Integer> heap = new PriorityQueue<>();
        for (Integer num : arr) {
            heap.add(num);
        }
        int num = 0;
        while (num < k) {
            result[num] = heap.poll();
            num++;
        }
        return result;
    }

}