卡码网编程基础课 | 8. 摆平积木

题目描述

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

输入描述

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

输出描述

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

输入示例

6
5 2 4 1 7 5
0

输出示例

5

python解法

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

C++解法

# include<iostream>
# include<vector>

int main(){
    int n;
    /************************ vector容器解法 *************************/
    while (std::cin >> n){
        if (n == 0) break;
        std::vector<int> nums = std::vector<int>(n, 0);
        int sum = 0;
        // 输入n堆积木每一堆的积木块数量，并计算总数量
        for (int i = 0; i < n; i++){
            std::cin >> nums[i];
            sum += nums[i];
        }
        // 计算平均值
        int average = sum / n;
         // 定义输出结果，表示需要移动的次数
        int result = 0;
        // 遍历每一摞积木
        for (int i = 0; i < n; i++){
            // 如果当前摞的积木超过平均值, 把超出平均值数量的积木移到不足平均值的地方，超出的数量就是需要挪动的次数
            if (nums[i] > average)  result += (nums[i] - average);
        }
        // 输出结果
        std::cout << result << std::endl;
        std::cout << std::endl;
    
    }
    
    /************************ 数组解法 *************************/
    // while (std::cin >> n){
    //     if (n == 0) break;
    //     int arr[n];
    //     int sum = 0;
    //     for (int i = 0; i < n; i++){
    //         std::cin >> arr[i];
    //         sum += arr[i];
    //     }
    //     int average = sum / n;
    //     int result = 0;
    //     for (int i = 0; i < n; i++){
    //         if (arr[i] > average) result += arr[i] - average;
    //     }
    //     std::cout << result << std::endl;
    //     std::cout << std::endl;
    // }
    
}