面试冲刺算法系列-43

输入n个整数，找出其中最小的K个数。例如输入4,5,1,6,2,7,3,8这8个数字，则最小的4个数字是1,2,3,4。

方法一：
将数组排序，取出前面K个值。

import java.util.Arrays;
import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        
        Arrays.sort(input);
        ArrayList<Integer> list = new ArrayList<Integer>();
        if(input == null || input.length<k)return list;
        for(int x=0;x<k;x++){
            list.add(input[x]);
        }
          
        return list;
    }
}

时间复杂度O(nlogn)
空间复杂度O(1)

方法二:
利用最大堆实现，先将前K个值放入最大堆，后K个值去跟堆顶元素进行比较，若小于堆顶，则替换掉堆顶，否则跳过。

import java.util.ArrayList;
import java.util.PriorityQueue;
import java.util.Comparator;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        ArrayList<Integer> result = new ArrayList<>();
        if(k > input.length||k == 0 || input == null)return result;
        PriorityQueue<Integer> queue = new PriorityQueue<Integer>(k,new Comparator<Integer>(){
        //大小堆是PriorityQueue的底层实现。
            @Override
            public int compare(Integer o1,Integer o2){
                return o2.compareTo(o1);
            }
        });
        for(int x=0;x<input.length;x++){
            if(queue.size()!= k){
                queue.offer(input[x]);
            }else{
                if(queue.peek()>input[x]){
                    Integer temp = queue.poll();
                    queue.offer(input[x]);
                    temp = null;
                }
            }
          
        }
       for(Integer s : queue){
           result.add(s);
       }
        return result;
    }
}