【卡码网Python基础课(7.摆平积木)】

---

题目描述与分析

1.描述

题目描述：
小明很喜欢玩积木。一天，他把许多积木块组成了好多高度不同的堆，每一堆都是一个摞一个的形式。然而此时，他又想把这些积木堆变成高度相同的。但是他很懒，他想移动最少的积木块来实现这一目标，你能帮助他吗？

输入描述：
输入包含多组测试样例。每组测试样例包含一个正整数n，表示小明已经堆好的积木堆的个数。
接着下一行是n个正整数，表示每一个积木堆的高度h，每块积木高度为1。其中1<=n<=50,1<=h<=100。
测试数据保证积木总数能被积木堆数整除。
当n=0时，输入结束。

输出描述：
对于每一组数据，输出将积木堆变成相同高度需要移动的最少积木块的数量。
在每组输出结果的下面都输出一个空行。

2.分析

题目要求我们把 n 堆高度不同的积木分成 n 堆高度相同的积木，假设我们面前真的有这样一堆积木，应该怎么做才能划分均等呢？

要想实现题目的要求，需要下面两步操作：

第一步：我们需要数清每一摞积木的数量，把他们的总数相加，这样我们就知道积木的总数量，再把得到的结果除以 n , 就得到了在高度相同的情况下，每一摞积木的块数。
第二步：对于超过平均值的积木，计算当前的积木数量和平均值的差值，把多的积木数量移到缺少的部分，直到积木高度相同。

代码编写

根据题目要求，输入包含多组测试样例，也就是说会有多个 n 的输入, 当输入 n 为0时，输入结束，我们可以写出下面的基础代码：

while True:
    try:
        # 变量n代表积木堆的个数
        n = int(input())
        # 当输入的n为0时，退出循环
        if n == 0:
            break
    except:
        break

积木是一组数据，可以使用上一节中学习的列表来进行处理，将输入的数据拆分成字符列表，然后使用map转为整数，最后转为列表。

# 将输入的一行数据转为整数列表
nums = list(map(int, input().split()))

列表计算总积木数量十分方便，直接使用sum()函数即可。

# 使用sum()对列表求和，获取总积木的个数
total_sum = sum(nums)

接着使用之前学习到的算术运算符计算 n 堆积木的平均值，为了使结果是整数，应该使用整除符号//

# 计算平均值， // 代表 整除
average = total_sum // n

对于每一摞积木，如果当前摞积木的数量超过平均值，也就是当前摞积木数量 > 平均值, 则把超出平均值数量的积木移到不足平均值的那一摞，超过的数量表示需要挪动的次数。

# result定义输出结果，表示积木需要挪动的次数
result = 0
# 遍历每一摞积木
for num in nums:
  # 如果当前摞的积木超过平均值, num表示当前摞的积木数量
    if num > average:
      # 则把超出平均值数量的积木移到不足平均值的地方，超出的数量就是需要挪动的次数
      # result = result + (nums - average)的简写形式
        result += num - average

最后按照题目要求，输出结果即可

# 输出结果
print(result)
print()

完整的代码如下：

while True:
    try:
        n = int(input())
        if n == 0:
            break
        # 获取每堆积木数量，将之转换成列表
        nums = list(map(int, input().split()))
        # 计算积木总数量
        total_sum = sum(nums)
        # 计算平均值
        average = total_sum // n
        # 定义输出结果
        result = 0
        # 遍历每一摞积木
        for num in nums:
        # 如果当前摞的积木超过平均值, 把超出平均值数量的积木移到不足平均值的地方，超出的数量就是需要挪动的次数
            if num > average:
                result += num - average
        # 输出结果
        print(result)
        print()
    except:
        break