牛客NC119 最小的K个数(优先队列或者排序)

题目

找前K小的数 维护一个长度为K的降序优先队列 排序这里就不写了
小于K直接入队，等于K时 如果当前元素小于队首元素，队首元素出队，当前元素入队

import java.util.ArrayList;
import java.util.PriorityQueue;
import java.util.*;
public class Solution {
    
    
    public ArrayList<Integer> GetLeastNumbers_Solution(int [] input, int k) {
    
    
       if(input.length == 0 || k > input.length || k == 0) return new ArrayList();
       PriorityQueue<Integer> queue = new PriorityQueue<Integer>((a,b) -> b-a);
       for(int i = 0; i < input.length; i++){
    
    
        if(queue.size() < k){
    
    
            queue.offer(input[i]);
        }else if(input[i] < queue.peek()){
    
    
                queue.poll();
                queue.offer(input[i]);
        }
    }
    ArrayList<Integer> res = new ArrayList<>();
        while(!queue.isEmpty()){
    
    
            res.add(queue.poll());
        }
            return res;
   }
}