求k个最小元素, 华为
Java 大根堆,维护k个最小的元素,时间复杂度O(NlogK)。
import java.util.*;
public class Main {
public static void main(String[] args){
Scanner sc = new Scanner(System.in);
while(sc.hasNext()){
int n = sc.nextInt(), k = sc.nextInt();
PriorityQueue<Integer> q = new PriorityQueue<>(Collections.reverseOrder());
for(int i=0; i < n; i++) {
int t = sc.nextInt();
if(q.isEmpty() || q.size() < k) q.offer(t);
else {
int min = q.peek();
if(t < min) {
q.poll();
q.offer(t);
}
}
}
ArrayList<Integer> res = new ArrayList<>();
while(!q.isEmpty()) res.add(q.poll());
for(int i=k-1; i > 0; i--)
System.out.print(res.get(i) + " ");
System.out.println(res.get(0));
}
}
}
【推荐】编程新体验,更懂你的AI,立即体验豆包MarsCode编程助手
【推荐】凌霞软件回馈社区,博客园 & 1Panel & Halo 联合会员上线
【推荐】抖音旗下AI助手豆包,你的智能百科全书,全免费不限次数
【推荐】轻量又高性能的 SSH 工具 IShell:AI 加持,快人一步