题目描述
有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)-> 从 ② 取11张牌放到①(10,10,10,10)。
输入格式
两行
第一行为:N(N 堆纸牌,1≤N≤100)
第二行为:A_1,A_2, … ,A_nN堆纸牌,每堆纸牌初始数,1≤Ai≤10000)
输出格式
一行:即所有堆均达到相等时的最少移动次数
寻找这样一个区间 (l ,r )只需要向一个方向移动卡牌,为此可能需要分割某个点。 这样的区间需要移动 r - l 次,否则就不符合区间的定义。
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37
| #include <bits/stdc++.h> using namespace std; typedef long long ll; #define sf(n) scanf("%d",&n) #define pf(n) printf("%d",n) #define pfc(c) printf(c) #define fi(i,s,t) for(int i=s;i<t;i++) #define fd(s,t) for(int i=s-1;i>=t;i--) #define mem(a,c) memset(a,c,sizeof(a)) const int INF=0x3f3f3f3f; const int MAX=100+5; int n,ans,a[MAX],cnt,cf,ce,avg=0; int main(){ freopen("in","r",stdin); sf(n); fi(i,0,n){ sf(a[i]); avg+=a[i]; } avg/=n; ans =0;cnt=0;cf=0,ce=-1; fi(i,0,n){ ce+=1; cnt +=a[i]; if(cnt >= (ce-cf+1) * avg){ ans += (ce-cf); if(i+1<n){ int t = cnt-(ce-cf+1) * avg; if(t) ans ++; a[i+1]+=t; cnt=0;cf=i+1;ce=i; } } } pf(ans); return 0; }
|