卡码网C++基础课 |7. 摆平积木

目录

前言

一、题目描述

二、解题思路

1.题目分析

2.代码编写

三、完整代码

总结

前言

仅个人记录所用

源自卡码网的C++基础课

“这门C++基础课 帮助 编程零基础学员快速学习刷算法题所需要的基础语法知识，学完之后，再来刷代码随想录，或者自己去刷编程题，基本不会有语法方面的困惑了，可以帮助大家快速上手刷题。 ”

第七题包含以下内容：

• vector的初始化操作练习
• vector的遍历和元素访问操作练习
• 计算平均值

---

一、题目描述

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

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

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

输入示例：

6

5 2 4 1 7 5

0

输出示例：

5

二、解题思路

1.题目分析

按照题目要求，将不同高度的积木移动成相同高度的积木，需要最少的移动方法。

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

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

2.代码编写

会有多组数据的输入，即n的输入很多，当n输入为0时，输入结束。基本代码如下：

#include<iostream>
using namespace std;
int main() {
  // 变量n代表积木堆的个数
  int n;
  // 多组测试数据，需要输入多个n
  while(cin >> n) {
    // 当输入的n为0时，退出循环
    if (n == 0) break;
  }
}

积木是一组数据，且每组数据的大小是动态变化的，最好使用vector。由不同的n，创建一个包含n个元素的容器，容器的默认值始终为0.

// 创建一个包含n个元素的容器，每个元素的值都是0
vector<int> nums = vector<int>(n, 0);

然后对vector进行遍历，将每个元素进行赋值，值的大小等于每一堆积木的积木块的数量，并计算总积木数量。

// sum代表总积木个数，初始化为0
int sum = 0;
for (int i = 0; i < n; i++) {
  // 输入n堆积木每一堆的积木块数量
    cin >> nums[i];
  // 对总数量进行累加
    sum += nums[i];
}

在循环种已经求得sum总和，再除以积木数量n，即可得到平均数。

那么对于每一摞积木，如果当前积木数量超过平均值，也就是当前积木数量-平均值>0，则可把差值当作移动数量。累计移动数量，即得到最终答案。

三、完整代码

#include<iostream>
#include<vector>
using namespace std;

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

---

总结

本节课练习了容器的遍历和元素访问操作，这是因为容器（数组）的遍历是经常使用的。

整体难度不大，但不看代码编写，仅自己思路去编写，特别容易犯小错误。比如每组数据都要清零，重新储存。如果在代码中写的push_back，那vector元素一定要清除。不然第二组数据就会出错。此外，还有n的问题，如果用的是while（n--），那后续就不能用n了，因为n已经减为0了，需要用nums.size()。其他可能还会有小错误，可以每个数据输出，找到问题所在。