小学生蓝桥杯Python闯关 | 石子传递

95 阅读1分钟

学习Python从娃娃抓起!记录下蓝桥杯Python学习和备考过程中的题目,记录每一个瞬间。

附上汇总贴:小学生蓝桥杯Python闯关 | 汇总_COCOgsta的博客-CSDN博客


【题目描述】

现有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