2558. 从数量最多的堆取走礼物

原题地址： 力扣每日一题：从数量最多的堆取走礼物

给你一个整数数组 gifts ，表示各堆礼物的数量。每一秒，你需要执行以下操作：

选择礼物数量最多的那一堆。
如果不止一堆都符合礼物数量最多，从中选择任一堆即可。
选中的那一堆留下平方根数量的礼物（向下取整），取走其他的礼物。
返回在 k 秒后剩下的礼物数量。

class Solution {public long pickGifts(int[] gifts, int k) {// 最大堆，从大到小维护优先队列// 堆里最后剩下的就是剩下的礼物数量// 1开方后还是1，无需进行特殊处理PriorityQueue<Integer> pri = new PriorityQueue<>((a, b) -> b - a);for (int gift : gifts) {pri.offer(gift);}while (k > 0) {k--;int x = pri.poll();pri.offer((int)Math.sqrt(x));}long res = 0;while (!pri.isEmpty()) {res += pri.poll();}return res;}
}

如果对您有帮助，请点赞关注支持我，谢谢！❤
如有错误或者不足之处，敬请指正！❤