<center>

问题 B: 【贪心】均分纸牌

时间限制: 1 Sec  内存限制: 64 MB
提交: 39  解决: 18
[提交][状态][讨论版]
</center>

题目描述

有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堆纸牌数(1<=N<=100)。
第二行为N堆纸牌中每堆纸牌初始数A1,A2,…,An(l<=Ai<=10000)。

输出

输出文件中仅一行,即所有堆均达到相等时的最少移动次数。

样例输入

4
9 8 17 6

样例输出

3
思路:
在网上看到的一个很好理解的解释。


代码:
#include <iostream>
#include <cstdio>

using namespace std;

int main()
{
    int n;
    int cou=0;
    int avg=0;
    int a[105];
    scanf("%d",&n);
    for(int i=0;i<n;i++){
        scanf("%d",&a[i]);
        avg+=a[i];
    }
    avg/=n;
    for(int i=0;i<=n-1;i++){
        if(a[i]==avg){
            continue;
        }else{
            if(a[i]<avg){
                a[i+1]-=(avg-a[i]);
                a[i]+=(a[i]-avg);
                cou++;
            }
            if(a[i]>avg){
                a[i+1]+=(a[i]-avg);
                a[i]-=(avg-a[i]);
                cou++;
            }
        }
    }
    printf("%d",cou);
    return 0;
}