题目描述 Description
有 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)。
输入描述 Input Description
第一行N(N 堆纸牌,1 <= N <= 100)
第二行A1 A2 … An (N 堆纸牌,每堆纸牌初始数,l<= Ai <=10000)
输出描述 Output Description
输出至屏幕。格式为:
所有堆均达到相等时的最少移动次数。‘
整体思路就是先计算n堆牌的平均数s,然后从第一堆开始向后遍历,用start记录起点的牌堆编号,同时用变量now记录start到i牌堆的总牌数,如果到第i堆时now<s*(i-start+1),那么说明这start到i堆需要从之外的牌堆拿牌,则继续向后遍历;若now=s*(i-start+1),则说明第i堆的牌比其应有的牌数多,则最后一堆得牌应该向前传,此时该组需要传递的次数就是i-start;若now>s*(i-start+1),则将多余的牌now-s*((i-start+1))向后传,然后再向前传,此时需要的次数是i-start+1。同时start=i+1,now=now-s*((i-start+1));
代码如下
#include<cstdio>
#include<iostream>
using namespace std;
int cards[101];
int main() {
int start=1,now=0,ans=0,s=0,n;scanf("%d",&n);
for(int i=1;i<=n;i++) {
scanf("%d",&cards[i]);
s+=cards[i];
}
s/=n;
for(int i=1;i<=n;i++) {
now+=cards[i];
if(now==s*(i-start+1)) {
ans+=i-start; start=i+1; now=0;
}
if(now>s*(i-start+1)) {
ans+=i-start+1;
now=now-s*(i-start+1); start=i+1;
}
}
printf("%d",ans);
return 0;
}