import java.util.*;

public class Solution {

    //堆的使用、lambda表达式的使用、迭代器的使用

    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {

        ArrayList<Integer> res = new ArrayList<>();

        if(input.length == 0 || k == 0) {

            return res;

        }

        //怎么判断是大根堆还是小根堆，目的是大根堆

        PriorityQueue<Integer> queue = new PriorityQueue<>((a, b) -> b - a);

        for(int i = 0; i < input.length; i++) {

            if(queue.size() < k) {

                queue.add(input[i]);

            } else if(queue.peek() > input[i]) {

                queue.poll();

                queue.add(input[i]);

            }

        }

        Iterator<Integer> iterator = queue.iterator();

        while(iterator.hasNext()) {

            res.add(iterator.next());

        }

        return res;

    }

}