卡码网语言基础课 | 摆平积木

题目分析

第一步：需要数清每一摞积木的数量，获取所有积木的总数，再把这个总数除以 n ，这样就得到了高度相同的情况下每一摞积木的数量（平均值）；

第二步：对于超过平均值的积木摞，计算当前积木的数量与平均值的差值，把多的积木移动到低于平均值的积木摞上，直到所有的积木摞都达到平均值。

代码实现

根据题目信息，可以写出如下基础代码：

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

积木是一组数据，可以使用上一节中学习的vector来进行处理，输入的 n 代表积木堆的个数, 因此可以创建一个包含 n 个元素的容器，容器的默认值为0。

vector<int> nums = vector<int>(n, 0);

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

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

计算n堆积木的平均值:

int average = sum / n;

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

int result = 0;  // result定义输出结果，表示积木需要挪动的次数
for (int i = 0; i < n; i++) {
	if ((nums[i] - average) > 0) {
		result += (nums[i] - average);
	}
}

最后输出结果：

cout << result << endl;
cout << endl;

完整代码如下：

#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;
		
		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;
	}
}