算法|每日一题|从数量最多的堆取走礼物|最大堆

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;
    }
}

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