前往小程序,Get更优阅读体验!
立即前往
首页
学习
活动
专区
工具
TVP
发布
社区首页 >专栏 >最小的K个数

最小的K个数

作者头像
名字是乱打的
发布2022-05-13 09:29:13
3010
发布2022-05-13 09:29:13
举报
文章被收录于专栏:软件工程

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

思路:

遍历数组先取数组的前k个数建立大根堆,继续遍历数组.如果当前值比大根堆最大值小,那么将数组中这个小值替换到堆里.

代码:

代码语言:javascript
复制
 //输入n个整数,找出其中最小的K个数。
    // 例如输入4,5,1,6,2,7,3,8这8个数字,则最小的4个数字是1,2,3,4,。
    public static ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> list=new ArrayList<Integer>();
        if (input==null){
            return list;
        }
        if (input.length<k||k==0){
            return list;
        }
        PriorityQueue<Integer> maxHeap=new PriorityQueue<Integer>(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2.compareTo(o1);
            }
        });
        //2,41,5,2,5,3,4,23
        for(int i=0;i<input.length;i++){
            if (i<k){
                maxHeap.add(input[i]);
            }
            else  if (maxHeap.peek()>input[i]){
                maxHeap.poll();
                maxHeap.add(input[i]);
                }
            }


        while (!maxHeap.isEmpty())
        {
            list.add(maxHeap.poll());
        }
        return list;
    }
本文参与 腾讯云自媒体同步曝光计划,分享自作者个人站点/博客。
原始发表:2022-05-13,如有侵权请联系 cloudcommunity@tencent.com 删除

本文分享自 作者个人站点/博客 前往查看

如有侵权,请联系 cloudcommunity@tencent.com 删除。

本文参与 腾讯云自媒体同步曝光计划  ,欢迎热爱写作的你一起参与!

评论
登录后参与评论
0 条评论
热度
最新
推荐阅读
目录
  • 思路:
  • 代码:
领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档