【问题描述】
直线上有n个等距离的村庄,每个村庄要么买酒,要么卖酒。设第i个村庄对酒的供给量为a[i],(显然a[i]>0表示该村庄供过于求,应该卖酒)。所有村庄供需平衡,即所有a[i]之和等于0。
把k个单位的酒从一个村庄运到另一个村庄需要k个单位的劳动力。计算最少需要多少劳动力可以满足所有村庄的需求。输出保证在64位带符号整数的范围内。
【输入格式】
第一行一个整数n,表示村庄的数量。接下来的n行,每行一个整数,第i+1行的整数表示a[i]。
【输出格式】
一个整数,表示最少的运费。
【输入样例】
【样例1】
5
5 -4 1 -3 1
【样例2】
6
-1000 -1000 -1000 1000 1000 1000
【输出样例】
【样例1】
9
【样例2】
9000
【数据范围】
2<=n<=100000, -1000<=a[i]<=1000
【思路梳理】
非常简单的一个贪心思路:对于每一个村庄而言,要使它的总运载费用最小,就应该让尽可能多的wine交易发生在他和与他尽可能近的村庄中。
由此一来,我们只需要从左往右依次扫描即可,每扫描到一个村庄的wine供给量不等于0,我们就把这个村庄的wine调至0(要么运出给下一个村庄,要么从下一个村庄运过来),再讲费用累加即可。
【Cpp环境】
#include<cstdio>
#include<iostream>
#include<cstring>
#define maxn 100005
using namespace std;
long long a[maxn],ans=0,tot=0;
int n;
long long read()//手工输入,读者可以不用管,防止极端情况下ans特别大超过int上界
{
long long cnt=0;
char ch;bool ok=false;
ch=getchar();
while((ch>'9'|| ch<'0')&& ch!='-') ch=getchar();
while((ch<='9' && ch>='0')|| ch=='-')
{
if(ch=='-') ok=true;
else cnt=cnt*10+ch-'0';
ch=getchar();
}
return ok==true? -cnt:cnt;
}
int main()
{
scanf("%d",&n);
for(int i=1;i<=n;i++) a[i]=read(),tot+=a[i];
for(int i=1;i<=n;i++)if(a[i])
{
ans+=a[i]>0? a[i]:-a[i];//等价于ans+=abs(a[i])
a[i+1]+=a[i];
a[i]=0;
}
cout<<ans;
return 0;
}