【贪心】均分纸牌

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

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

题目描述

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