题解 | #输入n个整数,输出其中最小的k个#

输入n个整数,输出其中最小的k个

http://www.nowcoder.com/practice/69ef2267aafd4d52b250a272fd27052c

import java.util.*;

public class Main {

public static void main(String[] args) {
    Scanner in = new Scanner(System.in);
    //最大的k个->小根堆,最小的k个->大根堆
    PriorityQueue<Integer> pq = new PriorityQueue<>((x, y) -> (y - x));
    while (in.hasNextInt()) {
        int n = in.nextInt(), k = in.nextInt();
        int[] nums = new int[n];
        //用数组承接输入整数
        for (int i = 0; i < n; i++) {
            nums[i] = in.nextInt();
        }
        //利用大根堆特性将最小的k个存入
        for(int num : nums){
            if(pq.size() == k && pq.peek() > num) pq.poll();
            if(pq.size() < k) pq.offer(num);
        }
        //用数组将大根堆中的元素倒序存入
        int[] res = new int[pq.size()];
        for(int i = res.length - 1; i >= 0; i--) res[i] = pq.poll();
        //从小到大输出
        for(int i = 0; i < k; i++){
            if(i == k - 1) System.out.print(res[i]);
            else System.out.print(res[i] + " ");
        }
    }
} 

}

全部评论

相关推荐

程序员小白条:你不是有一段实习了吗,现在找中大厂实习?过段时间要秋招了
我的简历长这样
点赞 评论 收藏
分享
门口唉提是地铁杀:之前b站被一个游戏demo深深的吸引了。看up主页发现是个初创公司,而且还在招人,也是一天60。二面的时候要我做一个登录验证和传输文件两个微服务,做完要我推到github仓库,还要我加上jaeger和一堆运维工具做性能测试并且面试的时候投屏演示。我傻乎乎的做完以后人家跟我说一句现在暂时不招人,1分钱没拿到全是白干
你的秋招第一场笔试是哪家
点赞 评论 收藏
分享
后来123321:别着急,我学院本大二,投了1100份,两个面试,其中一个还是我去线下招聘会投的简历,有时候这东西也得看运气
无实习如何秋招上岸
点赞 评论 收藏
分享
不愿透露姓名的神秘牛友
06-29 17:30
点赞 评论 收藏
分享
评论
点赞
收藏
分享

创作者周榜

更多
牛客网
牛客网在线编程
牛客网题解
牛客企业服务