题意:直线上有n个等距的村庄,每个村庄要么买酒要么卖酒,把k个单位的酒从一个村庄运送到另一个村庄需要k个单位的劳动力,问最少需要多少劳动力。
分析:考虑最左边的村庄,它的需求是a[i],那么它需要向第二个村庄转移abs(a[i])个需求,第二个村子现在需求为a[1]+a[2],再转移abs(a[1]+a[2])...以此类推累加答案即可。
#include<iostream>
#include<string>
#include<algorithm>
#include<cstdlib>
#include<cstdio>
#include<set>
#include<map>
#include<vector>
#include<cstring>
#include<stack>
#include<cmath>
#include<queue>
#define INF 0x3f3f3f3f
using namespace std;
long long n,a[100005];
int main()
{
cin.sync_with_stdio(false);
while(cin>>n && n)
{
long long tot = 0,ans = 0;
for(int i = 1;i <= n;i++)
{
cin>>a[i];
tot += 1ll*a[i];
ans += std::abs(tot);
}
cout<<ans<<endl;
}
}