2208. 将数组和减半的最少操作次数

2208. 将数组和减半的最少操作次数

这道题目难度不高，小白可能通过暴力遍历也可以做出来。O(N²）
但是如果使用优先级队列，也就是大根堆。时间复杂度会极大的降低。O(NlogN)

0x1 题目

给你一个正整数数组 nums 。每一次操作中，你可以从 nums 中选择 任意 一个数并将它减小到 恰好 一半。（注意，在后续操作中你可以对减半过的数继续执行操作）

请你返回将 nums 数组和 至少 减少一半的 最少 操作数。

• 地址：https://leetcode.cn/problems/minimum-operations-to-halve-array-sum/description/

0x2 示例

输入：nums = [5,19,8,1]
输出：3
解释：初始 nums 的和为 5 + 19 + 8 + 1 = 33 。
以下是将数组和减少至少一半的一种方法：
选择数字 19 并减小为 9.5 。
选择数字 9.5 并减小为 4.75 。
选择数字 8 并减小为 4 。
最终数组为 [5, 4.75, 4, 1] ，和为 5 + 4.75 + 4 + 1 = 14.75 。
nums 的和减小了 33 - 14.75 = 18.25 ，减小的部分超过了初始数组和的一半，18.25 >= 33/2 = 16.5 。
我们需要 3 个操作实现题目要求，所以返回 3 。
可以证明，无法通过少于 3 个操作使数组和减少至少一半。

0x3 解题思路

• 因为题目上的要求，至少和最少。所以，我们每次减半的数据应该都是数组中的最大值。接触过算法，我们很容易想到大根堆，这一堆结构。该堆结构可以让我们在O(logN)的时间内得到最大值。找对最大值后，将其减半，并将简单的数据进行累加。当累加数据大小达到原数组累积和一半时即可。
• 代码

  public int halveArray(int[] nums) {
  //	获取大根堆，默认是小根堆，需要更改比较方式改为大根堆
      PriorityQueue<Double> heap = new PriorityQueue<Double>((a, b) -> b.compareTo(a));
      for (int num : nums) {
          heap.add((double) num);
      }
      int res = 0;
      double sum = 0;
      for (int num : nums) {
          sum += num;
      }
      double sum2 = 0.0;
      while (sum2 < sum / 2) {
          double x = heap.poll();
          sum2 += x / 2;
          heap.add(x / 2);
          res++;
      }
      return res;
  }