最小的K个数

题目描述

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

用最大堆保存这k个数，每次只和堆顶比，如果比堆顶小，删除堆顶，新数入堆。

 

public ArrayList<Integer> GetLeastNumbers_Solution(int[] input, int k) {
  ArrayList<Integer> result = new ArrayList<>();
  int length = input.length;
  if(k > length || k == 0){
    return result;
  }
  PriorityQueue<Integer> maxHeap = new PriorityQueue<>(k, new Comparator<Integer>() {
    public int compare(Integer o1, Integer o2) {
      return o2.compareTo(o1); // 降序排列，堆顶是最大值，poll()会先取出最大值
    }
  });
  // lambda写法
  PriorityQueue<Integer> maxHeap = new PriorityQueue<>(k, (o1, o2) -> o2.compareTo(o1));

  for (int i = 0; i < length; i++) {
    if (maxHeap.size() != k) {
      maxHeap.offer(input[i]);
    } else if (maxHeap.peek() > input[i]) {
      maxHeap.poll();
      maxHeap.offer(input[i]);
    }
  }
  // 这里遍历的顺序就是maxHeap.poll()的顺序，第一个元素是最大值，第二个元素是第二大的
  for (Integer integer : maxHeap) {
    result.add(integer);
  }
  return result;
}