【算法】累加和减半最少操作数

题目链接

题目链接

题目描述

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

示例1

输入：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 个操作使数组和减少至少一半。

思路

1. 我们可以使用类似贪心的思想，如果想要尽可能的减少数组和的值，每次只能操作一个数，那么优先肯定从最大值开始减半，这样的操作数一定是最少；
2. 准备一个大根堆，然后每次将堆顶的元素减半，然后进行记录是否减少一半，减少完成后输出答案即可。

时间复杂度

O(N * logN)

题解

class Solution {
    
    static int size;
    static int N = 100010;
    static long[] heap = new long[N];

    public int halveArray(int[] nums) {
        size = nums.length;

        long sum = 0;
        for (int i = size - 1; i >= 0; i -- ) {
            heap[i] = (long) nums[i] << 20;
            sum += heap[i];
            down(i);
        }

        sum /= 2;
        int ans = 0;
        for (long min = 0; min < sum; ans ++) {
            heap[0] /= 2;
            min += heap[0];
            down(0);
        }
        return ans;
    }

    static void down(int i) {
        int l = i * 2 + 1;

        while (l < size) {
            int index = l + 1 < size && heap[l + 1] > heap[l] ? l + 1 : l;
            int b = heap[index] > heap[i] ? index : i;
            if (b == i) break;
            swap(heap, b, i);
            i = index;
            l = i * 2 + 1;
        }
    }

    static void swap(long[] q, int i, int j) {
        long t = q[i];
        q[i] = q[j];
        q[j] = t;
    }
}