剑指offer 40：最小的k个数

题目描述：

给定一个长度为 n 的可能有重复值的数组，找出其中不去重的最小的 k 个数。例如数组元素是4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4(任意顺序皆可)。

数据范围：0≤k,n≤10000，数组中每个数的大小0≤val≤1000

要求：空间复杂度O(n) ，时间复杂度O(nlogn)

示例1

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

返回值：[1,2,3,4]

说明：返回最小的4个数即可，返回[1,3,2,4]也可以

示例2

输入：[1],0

返回值：[]

示例3

输入：[0,1,2,1,2],3

返回值：[0,1,1]

解法一：排序

思路：

对数组进行排序，然后找到最小的k个数即可。

代码：

import java.util.*;

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

解法二：堆排序

思路：

        准备k个数字，之后每次遇到一个数字能够快速的与这k个数字中的最大值进行比较，然后将最大的之替换掉，那么最终剩下的就是最小的k个数字了。因此就可以使用优先级队列（大根堆），只要限制堆的大小为k，那么堆顶元素就是这k个数字中的最大值。如果需要进行替换，就将这个最大值拿出来，加入新的元素就好了。

1. 利用数组中的前k个数字构建一个大小为k的大根堆，堆顶元素为这k个元素的最大值
2. 遍历数组，依次比较其与堆顶元素的大小。若是比堆顶小，则弹出堆顶，再将新数加入堆中，直到数组结束。
3. 将堆顶依次弹出，即得到结果。

代码：

import java.util.*;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> res = new ArrayList<Integer>();
        //排除特殊情况
        if(k == 0 || input.length == 0) 
            return res;
        //大根堆 
        PriorityQueue<Integer> q = new PriorityQueue<>((o1, o2)->o2.compareTo(o1));
        //构建一个k个大小的堆  
        for(int i = 0; i < k; i++)
            q.offer(input[i]);
        for(int i = k; i < input.length; i++){
            //较小元素入堆
            if(q.peek() > input[i]){  
                q.poll();
                q.offer(input[i]);
            }
        }
        //堆中元素取出入数组
        for(int i = 0; i < k; i++) 
            res.add(q.poll());
        return res;
    }
}