【每日一题Day361】LC2558从数量最多的堆取走礼物 | 大顶堆

从数量最多的堆取走礼物【LC2558】

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

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

返回在 k 秒后剩下的礼物数量*。*

• 思路

  使用大顶堆存放所有礼物，每次将堆顶元素移出，并将其平方根放入堆中，最后求出堆中剩余礼物数目

• 实现

  class Solution {public long pickGifts(int[] gifts, int k) {long res = 0L;PriorityQueue<Integer> pq = new PriorityQueue<>((o1, o2) -> o2 - o1);for (int gift :gifts){pq.add(gift);res += gift;}while(k-- > 0){int poll = pq.poll(), left = (int)Math.sqrt(poll);res -= poll - left;pq.add(left);}return res;}
  }
  

  • 复杂度
    • 时间复杂度：$O(n+klogn)$
    • 空间复杂度：$O(n)$