最小的K个数

最小的K个数

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

分析：最直观的就是排后取出就行,复杂度为nlogn, 另一种解法是基于快排的partition函数,但是必须修改原数组才行,复杂度为n.目前用的这种解法用了辅助结构,用了优先级队列PriorityQueue来实现一个最大堆,也可以采用TreeSet基于红黑树的结构,复杂度nlogk.
PriorityQueue默认是最小堆,要注意用Collections.reverseOrder()可以产生一个比较器实现最大堆.
另外一点要特别注意k<=0时,和input.length < k时的特殊处理.

import java.util.PriorityQueue;
import java.util.Collections;
import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
        if (input.length < k || k <= 0) return new ArrayList<Integer>();
        PriorityQueue<Integer> priorityQueue = new PriorityQueue<>(k, Collections.reverseOrder());
        for (int i = 0; i < k; ++i) priorityQueue.offer(input[i]);
        for (int i = k; i < input.length; ++i) {
            if (input[i] < priorityQueue.peek()) {
                priorityQueue.poll();
                priorityQueue.offer(input[i]);
            }
        }
        ArrayList<Integer> result = new ArrayList<>();
        while (!priorityQueue.isEmpty()) {
            result.add(priorityQueue.poll());
        }
        return result;
    }
}