453. 最小移动次数使数组元素相等
给定一个长度为 n 的非空整数数组,找到让数组所有元素相等的最小移动次数。每次移动可以使 n - 1 个元素增加 1。
示例:
输入:
[1,2,3]
输出:
3
解释:
只需要3次移动(注意每次移动会增加两个元素的值):
[1,2,3] => [2,3,3] => [3,4,3] => [4,4,4]
思路
题目说每次都将余下n-1个元素加一,其实最优为留下最大的那个,其余n-1个数都加一,反着想就是最大那个数减一;所以最少的次数就相当于将这个list画成柱形图然后按最低的高度削平它,削掉的数量就是最少次数。
如果按照题目的正向意思;来做是会超时的,第一个程序就是如此。
正向想法
class Solution:
def minMoves(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
cnt = 0
while min(nums) != max(nums):
cnt += 1
nums = [x+1 for x in nums]
nums[nums.index(max(nums))] -= 1
print(nums)
return cnt
逆向想法
class Solution:
def minMoves(self, nums):
"""
:type nums: List[int]
:rtype: int
"""
min_index = min(nums)
cnt = sum(nums) - len(nums)*min_index
return cnt