[剑指offer]最小的K个数

叽歪丶

于 2018-12-16 11:10:10 发布

阅读量124

点赞数

CC 4.0 BY-SA版权

分类专栏：编程题文章标签：剑指offer

本文链接：https://blog.youkuaiyun.com/qq_37344518/article/details/85029026

编程题专栏收录该内容

40 篇文章

订阅专栏

本文介绍了一种高效算法，用于从输入数组中找到最小的K个数。通过使用最大堆，该算法能够快速筛选并返回所需数值。适用于大数据集处理，确保性能与准确性。

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

在这里插入图片描述
思路：
使用优先级队列（最大堆）保存这K个数，每次只和堆顶比，如果比顶堆小就删除堆顶，新数入堆，在堆大小等于K时就不用加入数了，直接将堆中的数都add进list中
实现：

import java.util.ArrayList;
import java.util.Comparator;
import java.util.PriorityQueue;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list = new ArrayList<>();
        int len = input.length;
        if(k > len || k == 0){
            return list;
        }
        PriorityQueue<Integer> maxHeap = new PriorityQueue<Integer>(k,new Comparator<Integer>(){
            public int compare(Integer o1,Integer o2){
                return o2.compareTo(o1);
            }
        });
        for(int i = 0;i < len ;i++){
            if(maxHeap.size() != k){
                maxHeap.offer(input[i]);
            }
            else if(maxHeap.peek() > input[i]){
                Integer temp = maxHeap.poll();
                temp = null;
                maxHeap.offer(input[i]);
            }
        }
        for(Integer integer : maxHeap){
            list.add(integer);
        }
        return list;
    }
}