移动次数最少（2007慈溪）C++代码

移动次数最少（2007慈溪）

有n堆糖果（2≤n≤200），排成一行，编号分别为1,2,…n。 已知每堆糖果有一定的颗数，且颗数之和均为n的倍数。移动各堆中的任意颗糖果，使每堆的数量达到相同，且移动次数最少。 移动规则： 每次可以移动任意的糖果颗数，第1堆可以移向第2堆，第2堆可以移向第1堆或第3堆,。。。。。。 第n 堆只可以移向第n -1堆。 例如，当n=4时： 堆号 1 2 3 4 颗数 9 8 17 6 移动的方法有许多种， 其中的一种方案： ① 第3堆向第4堆移动4颗，成为：9 8 13 10 ② 第3堆向第2堆移动3颗，成为：9 11 10 10 ③ 第2堆向第1堆移动1颗，成为：10 10 10 10 经过三次移动，每堆都成为10颗。

输入格式：
输入有两行。 第一行一个整数n。 第二行n个整数，用空格分隔。

输出格式：
输出一个整数（表示最少移动次数）

输入样例：

4
9 8 17 6

输出样例：

3

代码：

#include<iostream>
using namespace std;
int s[200]; // 前n项和 
int summ = 0;
int main() {
	int n, avg, l, key, cnt = 0;
	int a[200];
	cin >> n;
	for (int i = 0; i < n; i++) {
		cin >> a[i];
		summ += a[i];
		s[i] = summ;
	}
	key = n-1;
	avg = summ / n;
	l = key - 1;
	while (l != -1) {
		if (a[key] == avg) {
			key--;
			l--;
			continue;
		}
		int leftNeed = (l + 1) * avg - s[l];
		a[key] -= leftNeed;
		a[l] += leftNeed;
		key = l;
		l--;
		cnt++;
	}
	cout << cnt << endl;
//	for (int i = 0; i < n; i++) {
//		cout << a[i] << " ";
//	}
	return 0;
}