题目描述
有 N 堆纸牌,编号分别为 1,2,…, N。每堆上有若干张,但纸牌总数必为 N 的倍数。可以在任一堆上取若于张纸牌,然后移动。
移牌规则为:在编号为 1 堆上取的纸牌,只能移到编号为 2 的堆上;在编号为 N 的堆上取的纸牌,只能移到编号为 N-1
的堆上;其他堆上取的纸 牌,可以移到相邻左边或右边的堆上。 现在要求找出一种移动方法,用最少的移动次数使每堆上纸牌数都一样多。 例如
N=4,4 堆纸牌数分别为:① 9 ② 8 ③ 17 ④ 6 移动3次可达到目的: 从 ③ 取 4 张牌放到 ④ (9 8 13 10)
-> 从 ③ 取 3张牌放到 ②(9 11 10 10)-> 从 ② 取 1 张牌放到①(10 10 10 10)。
输入格式
N(N 堆纸牌,1 <= N <= 100)
A1 A2 … An (N 堆纸牌,每堆纸牌初始数,l<= Ai <=10000)
输出格式
所有堆均达到相等时的最少移动次数。
a[i]-=p
之后 a[i]
指离平均牌数(相等牌数)的数量,
一直向中间靠拢即可。
#include <iostream>
using namespace std;
int n, p = 0, ans = 0, l, r, a[10001];
int main() {
cin >> n;
for (int i = 1; i <= n; i++) cin >> a[i], p += a[i];
p /= n;
for (int i = 1; i <= n; i++) a[i] -= p;
l = 1;
r = n;
while (a[l] == 0 && l < n) l++;
while (a[r] == 0 && r > 1) r--;
while (l < r) {
a[l + 1] += a[l];
a[l] = 0;
ans++, l++;
while (a[l] == 0 && l < r) l++;
}
cout << ans << endl;
}