题解 | #最小的K个数#

最小的K个数

https://www.nowcoder.com/practice/6a296eb82cf844ca8539b57c23e6e9bf

import java.util.*;


public class Solution {
    /**
     * 代码中的类名、方法名、参数名已经指定,请勿修改,直接返回方法规定的值即可
     *
     *
     * @param input int整型一维数组
     * @param k int整型
     * @return int整型ArrayList
     */
    public ArrayList<Integer> GetLeastNumbers_Solution (int[] input, int k) {
        // write code here
        

        ArrayList<Integer> res = new ArrayList<Integer>();
        //建立大根堆,建立优先队列使用lambda表达式 初始化方式 为后者-前者
        PriorityQueue<Integer> p = new PriorityQueue<Integer>((p1, p2)
                ->p2 - p1);
        if(k==0||input.length==0){
            return res;
        }
        //先存满k个元素
        for (int i = 0; i < input.length; i++) {
            if (p.size() < k) {
                p.offer(input[i]);
            } else {
                if (input[i] < p.peek()) {
                    p.poll();
                    p.offer(input[i]);
                }
            }
        }
        for(int num:p){
            res.add(num);
        }
        // Iterator<Integer> iterator = p.iterator();
        // while (iterator.hasNext()) {
        //     res.add(iterator.next());
        // }
       Collections.sort(res);
        return res;
    }
}

全部评论

相关推荐

喜欢喜欢喜欢:这是我见过最长最臭的简历
点赞 评论 收藏
分享
03-31 18:02
门头沟学院 Java
白日梦想家_等打包版:不要的哦佛给我
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客企业服务