[面试算法题]最小操作次数使数组元素相等

126 阅读2分钟

给你一个长度为 n 的整数数组,每次操作将会使 n - 1 个元素增加 1 。返回让数组所有元素相等的最小操作次数。 示例 1:

    输入:nums = [1,2,3]

    输出:3

    解释:

        只需要3次操作(注意每次操作会增加两个元素的值):

        [1,2,3] => [2,3,3] => [3,4,3] => [4,4,4] 示例 2:

    输入:nums = [1,1,1]

    输出:0 提示:

    n == nums.length

    1 <= nums.length <= 105

    -109 <= nums[i] <= 109

    答案保证符合 32-bit 整数

为了找到使数组所有元素相等所需的最小操作次数,我们可以观察到,每次操作实际上是除了一个元素外的所有元素都增加了1。这意味着,如果我们选择数组中的最小值作为目标值,那么我们可以将所有其他元素都减小到该最小值,而不是增加除了一个元素外的所有元素。

因此,我们可以找到数组中的最小值,然后计算每个元素需要减少多少次才能达到这个最小值。由于每次操作都可以使n-1个元素增加1(等价于除了一个元素外的所有元素增加1),所以我们可以将所有元素与最小值的差值累加起来,这个累加值就是所需的最小操作次数。

以下是实现这个逻辑的JavaScript代码:

function minMoves(nums) {
    // 找到数组中的最小值
    let minVal = Math.min(...nums);
    // 初始化操作次数为0
    let moves = 0;
    // 遍历数组,累加每个元素与最小值的差值
    for (let num of nums) {
        moves += num - minVal;
    }
    // 返回操作次数
    return moves;
}

// 示例测试
console.log(minMoves([1, 2, 3])); // 输出: 3
console.log(minMoves([1, 1, 1])); // 输出: 0

在这个实现中,我们使用Math.min(...nums)来找到数组中的最小值。然后,我们遍历数组并累加每个元素与最小值的差值。这个差值实际上表示了为了将该元素减小到最小值所需的操作次数(因为每次操作都可以让除了一个元素外的所有元素增加1,所以实际上等于让当前元素减少)。最后,我们返回累加的操作次数。