剑指Offer:29.最小的K个数

原创于 2019-07-21 13:38:40 发布 · 131 阅读

0 ·

CC 4.0 BY-SA版权

Java 同时被 3 个专栏收录

183 篇文章

订阅专栏

数据结构

154 篇文章

订阅专栏

剑指Offer提高班

67 篇文章

订阅专栏

本文介绍了一种使用最大堆寻找数组中最小K个数的方法。通过构造一个最大堆，可以高效地找到给定数组中最小的K个元素，并返回这些元素组成的列表。

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

题目描述
输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4,。

import java.util.*;
//思路：构造一个最小堆构成的最大堆
public class Solution {
    ArrayList<Integer> ans=new ArrayList<>();
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if(k>input.length||k<=0)return ans;
        //构造大顶堆
        PriorityQueue<Integer>pq=new PriorityQueue<>((o1,o2)->(o2-o1));
        for(int num:input){
            pq.offer(num);
            if(pq.size()>k)
                pq.poll();
        }
        return new ArrayList<>(pq);
    }
}