洛谷 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

这个题比较有意思//

思路：找出他们的平均数，用所有的数减去这个平均数，然后就开始加，每次加到0就跳过，否则次数加1；

#include <bits/stdc++.h>
typedef long long ll;
using namespace std;
ll a[10005];
int main()
{
    ios::sync_with_stdio(false);
    ll n;
    ll s=0;
    cin>>n;
    for(int i=0;i<n;i++)
    {
        cin>>a[i];
        s+=a[i];
    }
    s/=n;
    ll k=0;
    for(int i=0;i<n;i++)
    {
        a[i]-=s;
    }
    for(int i=0;i<n;i++)
    {
        if(a[i]==0)
            continue;
        else
            a[i+1]+=a[i],
            k++;
    }
    cout<<k<<endl;
    return 0;
}