石子传递

题目描述

现有N（1<N<50）堆石子，环形摆放。每堆石子的石子数量可由输入确定。现在要使N堆石子的数量变为相同（输入数据保证能使每堆石子数量相同），每次可以从任意一堆中石子中取出1个，放入到与之相邻的另一堆中。请编写一段程序，计算并输出最少需要移动的石子数量。如果原始N堆石子数量相同，则输出0。

输入描述

第一行输入一个正整数N（1<N<50），表示石子的堆数

第二行输入N个正整数（1<正整数<100），表示顺时针方向每堆石子的原始数量，正整数之间以一个空格隔开

输出描述

输出一个整数，表示要使N堆石子的数量变为相同，最少移动的石子数量，如果原始N堆石子数量相同，则输出0

样例输入

3

2 6 7

样例输出

3

代码详解

展开查看

n = int(input())
ls = [int(i) for i in input().split()]
avg = sum(ls) // n

for i in range(n):
    if i == 0:
        ls[i] -= avg
    else:
        ls[i] -= avg
        ls[i] += ls[i-1]

ls.sort()
tot = 0
for x in ls:
    tot += abs(x - ls[n // 2])

print(tot)

运行结果

展开查看

3
2 6 7
3