剑指Offer:最小的K个数 (java代码实现)

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

第一思路 先排序然后找出前k个数 这种算法复杂度和排序的时间复杂度一样 最优为O(nlogn) 下面介绍一种方法 时间复杂度为O(nlogk);

解题思路
1.维护一个大根堆 遍历数组先放k个数进堆 堆顶的元素就是这个k个元素中最大的
2.继续遍历 如果当前数大于堆顶元素 删除堆顶元素 当前元素入堆
遍历完数组后堆里的元素 就是k个最小的。

因为只需要维护k个元素的堆 所以该算法时间复杂度为0(nlogk);

下面是代码实现

import java.util.PriorityQueue;
import java.util.Comparator;
import java.util.ArrayList;
public class Solution {
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
      ArrayList<Integer> list = new ArrayList();
        if (k > input.length || k == 0) {
            return list;
        }
        PriorityQueue<Integer> queue = new PriorityQueue(new Comparator<Integer>() {
            @Override
            public int compare(Integer o1, Integer o2) {
                return o2 - o1;
            }
        });

        for (int i = 0; i < input.length; i++) {
            if (i < k) {
                queue.offer(input[i]);
            }else if(queue.peek()>input[i]) {
                queue.poll();
                queue.offer(input[i]);
            }
        }

        for (int i = 0; i < k; i++) {
            list.add(queue.poll());
        }
        return list;
    }
}