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

 

思路：求出平均数，求出每个数与平均数的差，等于0的不用管，不等于0的加到下一个数上

#include<bits/stdc++.h>
using namespace std;
const int maxn=10008;
int a[maxn];
int main()
{
    int average,n,sum=0,count1=0;
    scanf("%d",&n);
    for(int i=0;i<n;i++)
    {
      scanf("%d",&a[i]);
      sum+=a[i];
    }
    average=sum/n;
    for(int i=0;i<n;i++)
    a[i]-=average;
    for(int i=0;i<n-1;i++)
    {
      if(a[i]==0)
       continue;
      a[i+1]+=a[i];
      count1++;
    }
    printf("%d",count1);
    return 0;
}

说明

对于20%的数据，n≤10；
对于50%的数据，n≤1000；
对于70%的数据，n≤100000；
对于100%的数据，n≤1000000，0≤ai＜bi≤1000000。