go语言题解LeetCode453最小操作次数使数组元素相等

题目描述

原题链接 :

453. 最小操作次数使数组元素相等 - 力扣(LeetCode) (leetcode-cn.com)

给你一个长度为 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 <= 10^5

-10^9 <= nums[i] <= 10^9

答案保证符合 32-bit 整数

思路分析

首先换个角度理解题目,将数组按顺序表示为高度为nums[i]宽度为1的柱子,每次移动i的含义为除i外的所有柱子高度加1,这样i柱子与其他柱子的相对高度减1,可将移动i理解为将i柱子高度减1,其他柱子不变。于是题目变为,求将所有柱子减少为高度相等的柱子所需移动次数。由于每次都只能减少某根柱子的高度1,所以最少要移动sum(nums[i]-min(nums))次。

在遍历nums的过程中,如果nums[i]大于之前的最小值,则nums[i]-min为需要减少的高度;

如果nums[i]小于之前的最小值,则需把前面没有减少的高度i*(min-nums[i]),一次减掉。

AC 代码

func minMoves(nums []int) int {
    min := 1<<31 - 1
    sum := 0
    for i := range nums {
        if nums[i] >= min {
            sum += (nums[i]-min)
        } else {
            sum += i * (min-nums[i])
            min = nums[i]
        }
    }
    return sum
}

小结

这道题主要理解题意之后,转换一下就简单了。

只需遍历数组一次即可 - 最小操作次数使数组元素相等

遍历数组一次

解题思路

对于要使得数组所有元素相等这个最终目标而言,将n-1个元素加一其实等价于将剩下的那个元素减一,因此只需要将所有元素进行减一操作直到等于数组中最小元素即可。所以最终的答案就是所有元素与最小元素的差值之和。

代码

class Solution {
    public int minMoves(int[] nums) {
        if (nums.length<=1)
            return 0;
        int sum=0,min=Integer.MAX_VALUE;
        int base = nums[0];
        for (int num:nums){
            sum+=num-base;
            min=Integer.min(min,num);
        }
        return sum+(base-min)*nums.length;
    }
}

强行找规律

解题思路

等于排序数组除第一位以外的和减去数组长度减1乘以数组第一个数

代码

class Solution:
    def minMoves(self, nums: List[int]) -> int:
        nums.sort()
        return sum(nums[1:])-(len(nums)-1)*nums[0]

以上就是go语言题解LeetCode453最小操作次数使数组元素相等的详细内容,更多关于go最小操作数组元素相等的资料请关注其它相关文章!

原文地址:https://juejin.cn/post/7175059396739727418