【一次过】【优先队列】Lintcode 544. 前K大数

小马哥MAX

于 2018-07-19 10:29:32 发布

阅读量303

点赞数

CC 4.0 BY-SA版权

分类专栏： lintcode 优先队列

本文链接：https://blog.youkuaiyun.com/majichen95/article/details/81110894

lintcode 同时被 2 个专栏收录

393 篇文章

订阅专栏

优先队列

5 篇文章

订阅专栏

本文介绍了一种使用小根堆实现的高效算法来找出数组中的前K个最大值。该方法通过维护一个大小为K的小根堆，确保堆内始终保留最大的K个元素，最后返回这些元素并按要求顺序排列。

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

在一个数组中找到前K大的数

样例

给出 [3,10,1000,-99,4,100], k = 3.
返回 [1000, 100, 10]

解题思路：

看到取前X大(小)数，使用优先队列即可。

注意这里使用小根堆，大小始终保持K，不断poll最小的数，最终小根堆会只留下前K大的数

由于poll出来的数是由小到大，而题目要求由大到小，所以取数据时记得反取

import java.util.*;

public class Solution {
    /**
     * @param nums: an integer array
     * @param k: An integer
     * @return: the top k largest numbers in array
     */
    public int[] topk(int[] nums, int k) {
        //小根堆
        PriorityQueue<Integer> pq = new PriorityQueue<>();
        
        for(int i=0; i<nums.length; i++){
            pq.offer(nums[i]);
            if(pq.size() > k){
                pq.poll();
            }
        }
        
        int[] res = new int[k];
        for(int i=k-1; i>=0; i--)
            res[i] = pq.poll();
        
        return res;
    }
}