java: add «2558. Take Gifts From the Richest Pile»

URL:	https://leetcode.com/problems/take-gifts-from-the-richest-pile/
Signed-off-by: Matej Focko <me@mfocko.xyz>
This commit is contained in:
Matej Focko 2024-12-12 22:52:23 +01:00
parent a58d78d2eb
commit da959699dd
Signed by: mfocko
SSH key fingerprint: SHA256:icm0fIOSJUpy5+1x23sfr+hLtF9UhY8VpMC7H4WFJP8

View file

@ -0,0 +1,19 @@
class Solution {
public long pickGifts(int[] gifts, int k) {
var q = new PriorityQueue<Integer>(Comparator.reverseOrder());
for (var gift : gifts) {
q.offer(gift);
}
for (int i = 0; i < k; ++i) {
int richest = q.poll();
q.offer((int) Math.sqrt(richest));
}
long remaining = 0;
while (!q.isEmpty()) {
remaining += q.poll();
}
return remaining;
}
}