AcWing 53. 最小的k个数-java

AcWing 53. 最小的k个数

快速排序和最大堆的时间复杂度都是 nlogn

原题链接

输入 n 个整数，找出其中最小的 k 个数。

注意：

输出数组内元素请按从小到大顺序排序;
数据范围
1≤k≤n≤1000

代码案例：输入：[1,2,3,4,5,6,7,8] , k=4
输出：[1,2,3,4]

题解

1 暴力做法

最朴素的做法就是排序 然后输出

class Solution {
    public List<Integer> getLeastNumbers_Solution(int [] input, int k) {
        List<Integer>  res = new ArrayList<>();
        Arrays.sort(input);
        for(int i = 0 ; i < k ; i++){
             res.add(input[i]);
        }
        return res ;
    }
}

2、最大堆

维护一个大小为k的大根堆，将数组元素都push进堆，当堆中的数大于k时弹出堆顶元素。注意弹出堆顶的顺序是从大到小的k个数，要进行逆序操作
最大堆 把最大的数弄出去 剩下的就是最小的数 注意的是 输出是按照从大到小的顺序输出的 所以使用addFirst函数加入到list的头

class Solution {
    public List<Integer> getLeastNumbers_Solution(int [] input, int k) {
        
        LinkedList<Integer> list = new LinkedList<>();
        Queue<Integer> heap = new PriorityQueue<>((a, b)->(b - a));//最大堆 同时lamba表达式显示
        for(int i = 0 ; i < input.length ; i++){
            heap.offer(input[i]);
            if(heap.size() > k) heap.poll();
        }
        while(!heap.isEmpty()) list.addFirst(heap.poll());
        return list ;
    }
}