青藤 #10021 均分纸牌

题目描述

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