Advertisement
k largest elements
JavaView on GFG
k largest elements.java
Java
class Solution {
public ArrayList<Integer> kLargest(int[] arr, int k) {
PriorityQueue<Integer> pq = new PriorityQueue<>();
for (int num : arr) {
pq.offer(num);
if (pq.size() > k)
pq.poll();
}
ArrayList<Integer> ans = new ArrayList<>(Collections.nCopies(k, 0));
while (!pq.isEmpty())
ans.set(--k, pq.poll());
return ans;
}
}Advertisement
Was this solution helpful?