题目
给你一个长度为 n 的整数数组,每次操作将会使 n - 1 个元素增加 1 。返回让数组所有元素相等的最小操作次数。
示例:
输入:nums = [1,2,3]
输出:3
解释:
只需要3次操作(注意每次操作会增加两个元素的值):
[1,2,3] => [2,3,3] => [3,4,3] => [4,4,4]
输入:nums = [1,1,1]
输出:0
分析
- 思路1
对于该问题,直接统计“+1”操作次数可能会导致一些问题,如:nums中的最大值和最小值差值过大时,程序效率极低。
因此,可以考虑逆过程:每次将n-1个元素+1直到所有元素相等,等价于每次将1个元素-1直到所有元素相等。
实现时,累加nums中每个元素与最小值的差
- 思路2
来自宫水三叶大神的数学求解思路:
令原数组 nums 的总和为 sum,最小值为 min,最大值为 max,长度为 n,真实最小操作次数为 ans。
根据题目的描述,nums 最终状态的元素值 t 满足:t ≥ max,当且仅当 nums 中有 n-1 个元素与 max 差值相等时取等号。
ans 与 t 间的关系:
求最小的 ans 等价于求最小的 t,但是目前已知的仅有 t ≥ max
将问题抽象为补全水桶的短板,在 nums 中的元素到达 t 的过程中,min 每轮都会+1,因此可以得到一个关系:t = min + ans
带入得到一个新关系式:
变形化简得:
实现
思路1:
def minMoves(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
min_num = min(nums)
res = 0
for item in nums:
res += item - min_num
return(res)
思路2:
class Solution(object):
def minMoves(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
return(sum(nums) - min(nums) * len(nums))