有N堆纸牌,编号分别为 1,2,…,N。
每堆上有若干张,但纸牌总数必为 N 的倍数。
可以在任一堆上取若干张纸牌,然后移动。
移牌规则为:在编号为 1 的堆上取的纸牌,只能移到编号为 2 的堆上;在编号为 N 的堆上取的纸牌,只能移到编号为 N−1 的堆上;其他堆上取的纸牌,可以移到相邻左边或右边的堆上。
现在要求找出一种移动方法,用最少的移动次数使每堆上纸牌数都一样多。
例如 N=4,4 堆纸牌数分别为:(9,8,17,6)。
移动 3 次可达到目的:
从第三堆取四张牌放入第四堆,各堆纸牌数量变为:(9,8,13,10)。
从第三堆取三张牌放入第二堆,各堆纸牌数量变为:(9,11,10,10)。
从第二堆取一张牌放入第一堆,各堆纸牌数量变为:(10,10,10,10)。
输入格式
第一行包含整数 N。
第二行包含 N 个整数,A1,A2,…,AN 表示各堆的纸牌数量。
输出格式
输出使得所有堆的纸牌数量都相等所需的最少移动次数。
数据范围
1≤N≤100,
1≤Ai≤10000
输入样例:
4
9 8 17 6
输出样例:
3
思路分析
设纸牌平均数为a,ai代表第i堆纸牌的张数,xi代表第i堆纸牌向第i+1堆纸牌移动的纸牌数目。
则
a1 - x1 = a
a2 +x1 - x2 = a (第1堆牌向第2堆牌移动了x1张,第2堆牌向第3堆牌移动了x2张,移动后第2堆牌的数目等于纸牌平均数a)
…
an-1 + xn-2 - xn-1 = a
an + xn-1 = a
可递推每堆牌移动的数目。
x1=a1 - a
x2=a2 - a + x1
x3=a3 - a + x2
…
xn-1=an-1 - a + xn-2
具体方案递归策略:
- xi=0时,这堆牌无需移动。
- xi>0时,先从ai转移xi张牌给ai+1,然后再递归操作ai+1 到 an。
- xi<0时,先递归操作ai+1 到 an,然后再从ai+1转移|xi|张牌给ai。
而本题继只需计算出最少移动次数,即统计不为0的 xi的个数。
代码
#include<iostream>
#include<algorithm>
using namespace std;
int a[105];
int main(){
int n,sum=0;
cin>>n;
for(int i=0;i<n;i++){
cin>>a[i];
sum+=a[i];
}
int avg = sum / n;//求平均值
int res = 0;
for(int i =0,x=0;i<n;i++){//初始时可以看做别人给x1 0张牌,好统一处理递推
x = a[i] -avg + x;
if(x)//x不为0,需要移动牌
res++;
}
cout<<res<<endl;
return 0;
}