最后每个洗衣机都会得到均值。所以,对于单个的洗衣机,移动衣服的次数最小是均值减去洗衣机里的衣服数。同样,对于一组相邻的洗衣机,总的移动次数也最小是均值*洗衣机数字 减去 这堆洗衣机里的衣服的总数。
题目用greedy,可以维护三个变量,一个全局最大值,一个当前洗衣机的衣服量与均值的差距,一个是从开始到当前洗衣机位置的总和的移动次数。选择三者最大值。
class Solution:
def findMinMoves(self, machines: List[int]) -> int:
total = sum(machines)
N = len(machines)
if total % N :return -1
des = total // N
res = 0
group = 0
for ma in machines:
group += ma - des
res = max(res, abs(group), ma - des)
return res