剑指Offer--Java--最小的k个数

题目描述

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

注意：

数据保证k一定小于等于输入数组的长度;
输出数组内元素请按从小到大顺序排序;

样例描述

输入：[1,2,3,4,5,6,7,8] , k=4

输出：[1,2,3,4]

思路

一. 小根堆法 手动实现，直接升序排序(快速排序，堆排序…)，然后找最小的k个数。这里以堆排序为例，手动实现最小堆。 面试喜欢考堆排或者快排的手动实现。
二. 大根堆法 (更常用)，尤其对于海量数据，算法效率很高

1. 考虑用大根堆，这里用优先队列PriorityQueue来实现。维护一个最大根堆heap里面放k个数，设置(a, b) -> b.compareTo(a)使得队列里面是逆序排列的，则此时最大的数就是优先级最高的数，也就是使用poll()时弹出的数。优先队列具有自动排序功能，所以每次添加数时，保证了就是一个大根堆。从而弹出所有大的数后，剩余堆里就是最小的k个数。
2. PriorityQueue默认是升序，对基本数据类型来说，最小的元素是优先级最高的数。

代码

1. 小根堆法

class Solution {
    public List<Integer> getLeastNumbers_Solution(int [] input, int k) {
        List<Integer> ans=new ArrayList<Integer>();
        int len=input.length;
        if(k>len)
        return ans;
        for(int i=0;i<k;i++){
            heapSort(input,i,len);
            ans.add(input[i]);
        }
        return ans;
    }
    void heapSort(int[] input,int i,int len){
        for(int j=len-1;j>=i;j--){
            int p=(j+i-1)/2;
            if(input[p]>input[j]){
                int t=input[p];
                input[p]=input[j];
                input[j]=t;
            }
        }
    }
}

2. 大根堆法，尤其对于海量数据，算法效率很高

class Solution {
    public List<Integer> getLeastNumbers_Solution(int [] input, int k) {
        //最大堆，优先队列                      使得b>a!
        Queue<Integer> heap=new PriorityQueue<>((a, b) -> b.compareTo(a));
        for(int x:input){
            heap.add(x);
            if(heap.size()>k) heap.poll();
        }
        List<Integer> ans=new ArrayList<>();
        while(heap.size()>0){
            ans.add(heap.peek());
            heap.poll();
        }
        //逆序，由于里面的是最大堆，是从大到小出来的
        Collections.reverse(ans);
        return ans;
    }
}