题解 | #输入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] + " ");
        }
    }
} 

}

全部评论

相关推荐

不愿透露姓名的神秘牛友
11-21 17:16
科大讯飞 算法工程师 28.0k*14.0, 百分之三十是绩效,惯例只发0.9
点赞 评论 收藏
分享
10-17 12:16
同济大学 Java
7182oat:快快放弃了然后发给我,然后让我也泡他七天最后再拒掉,狠狠羞辱他一把😋
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务