最小的K个数

描述

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

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

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

首先想到的是使用堆排序：

这里使用arraylist模拟堆：

import java.util.ArrayList;
 
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> kList = new ArrayList<>();
        // 先写入k个数据
        for ( int i = 0;i< k;i++){
            Integer v = input[i];
            kList.add(v);
        }
         
        // 排序
        kList.sort((o1,o2)-> {return o1 - o2;});
        for ( int i = k;i< input.length;i++ ){
             
            Integer v = input[i];
            // 判断当前值和 堆的最大值 谁大，如果比堆的最大值大，忽略
            if(kList.size()>0){
                Integer last = kList.get(kList.size()-1);
                 
                if(last <  v){
                    continue;
                }
            }
             
            kList.add(v);
             
            kList.sort((o1,o2)-> {return o1 - o2;});
            kList.remove(kList.size()-1);
        }
        kList.sort((o1,o2)-> {return o1 - o2;});
        return kList;
    }
}

使用优先级队列替换arraylist：

import java.util.ArrayList;
import java.util.PriorityQueue;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
         ArrayList<Integer> res = new ArrayList<>(k);
    //根据题意要求，如果K>数组的长度，返回一个空的数组
    if (k > input.length || k == 0)
        return res;
    //创建最大堆
    PriorityQueue<Integer> queue = new PriorityQueue<>((num1, num2) -> num2 - num1);
    //先在堆中放数组的前k个元素
    for (int i = 0; i < k; ++i) {
        queue.offer(input[i]);
    }
    //因为是最大堆，也就是堆顶的元素是堆中最大的，遍历数组后面元素的时候，
    //如果当前元素比堆顶元素大，就把堆顶元素给移除，然后再把当前元素放到堆中，
    for (int i = k; i < input.length; ++i) {
        if (queue.peek() > input[i]) {
            queue.poll();
            queue.offer(input[i]);
        }
    }
    //最后再把堆中元素转化为数组
    for (int i = 0; i < k; ++i) {
        res.add(queue.poll());
    }
    return res;
    }
}