LeetCode——最小的k个数(Java)

最新推荐文章于 2022-04-22 08:28:21 发布

Rabbit_Sir

最新推荐文章于 2022-04-22 08:28:21 发布

阅读量158

点赞数 1

分类专栏： LeetCode 文章标签： queue

本文链接：https://blog.csdn.net/qq_45062721/article/details/105000788

版权

LeetCode 专栏收录该内容

134 篇文章 2 订阅

订阅专栏

直接用堆进行排序，然后进行转化就行。
在这里插入图片描述

 package LeetCode.剑指offer;

import java.util.PriorityQueue;
import java.util.Queue;

public class Fourth {
    public int[] getLeastNumbers(int[] arr, int k) {
        if (k == 0){
            return new int[0];
        }
        // 使用一个最大堆（大顶堆）
        // Java 的 PriorityQueue 默认是小顶堆，添加 comparator 参数使其变成最大堆
        Queue<Integer> queue = new PriorityQueue<>(k, (i1,i2) -> Integer.compare(i1,i2));

        //入堆
        for (int a : arr){
            //当前数字小于堆元素才会入堆
            if (queue.isEmpty() || queue.size() < k || a < queue.peek())
                queue.offer(a);
            if (queue.size() > k)
                queue.poll();
        }

        //将堆中元素加到数组中
        int [] sum = new int[k];
        int j = 0;
        for (int i : queue){
            sum[j++] = i;
        }
        return sum;
    }
}