洛谷 P1031： 均分纸牌

https://www.luogu.org/problemnew/show/P1031

题目描述

有NNN堆纸牌，编号分别为 1,2,…,N1,2,…,N1,2,…,N。每堆上有若干张，但纸牌总数必为NNN的倍数。可以在任一堆上取若干张纸牌，然后移动。

移牌规则为：在编号为111堆上取的纸牌，只能移到编号为222的堆上；在编号为NNN的堆上取的纸牌，只能移到编号为N−1N-1N−1的堆上；其他堆上取的纸牌，可以移到相邻左边或右边的堆上。

现在要求找出一种移动方法，用最少的移动次数使每堆上纸牌数都一样多。

例如N=4N=4N=4，444堆纸牌数分别为：

①999②888③171717④666

移动333次可达到目的：

从 ③ 取444张牌放到 ④ （9,8,13,109,8,13,109,8,13,10）-> 从 ③ 取333张牌放到 ②（9,11,10,109,11,10,109,11,10,10）-> 从 ② 取111张牌放到①（10,10,10,1010,10,10,1010,10,10,10）。

输入输出格式

输入格式：

两行

第一行为：NNN（NNN 堆纸牌，1≤N≤1001 le N le 1001≤N≤100）

第二行为：A1,A2,…,AnA_1,A_2, … ,A_nA1​,A2​,…,An​ （NNN堆纸牌，每堆纸牌初始数，l≤Ai≤10000l le A_i le 10000l≤Ai​≤10000）

输出格式：

一行：即所有堆均达到相等时的最少移动次数。

输入输出样例

输入样例#1： 复制

4
9 8 17 6

输出样例#1： 复制

3

先把平均值算出来，用一个b数组保存每个数给或拿周围的数的数量，如果为0，证明这个元素没动过。

#include<stdio.h>
#define N 120
int a[N],b[N];
int main()
{
    int n,i,s,sum=0,ans=0;
    scanf("%d",&n);
    for(i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        sum+=a[i];
    }
    s=sum/n;
    b[1]=s-a[1];
    for(i=2;i<n;i++)
    {
        b[i]=s+b[i-1]-a[i];
    }
    b[n]=s+b[i-1];
    for(i=1;i<n;i++)
        if(b[i])
            ans++;
        
    printf("%d
",ans);
    return 0;
}