4.11 村庄买卖酒

题目

直线上有n（2<=n<=100000)个等距的村庄，每个村庄要么买酒，要么卖酒。设第i个村庄对酒的需求为ai(-1000<=ai<=1000),其中ai>0表示买酒，ai<0表示卖酒。所有村庄供需平衡，即所有ai之和等于0。
把k个单位的酒从一个村庄运到相邻村庄需要k个单位的劳动力。计算最少需要多少劳动力可以满足所有村庄的需求（使得数组中所有数为0的最少移动量）。输出保证在64位带符号整数的范围内。

思路

考虑最左边的村庄。如果需要买酒，即a1>0，则一定有劳动力从村庄2往左运给村庄1，而不管这些酒是从哪里来的（可能就是村庄2产的，也可能谁更右边的村庄运到村庄2的）。这样，问题就等价只有村庄2~n，且第二个村庄的需求为a1+a2的情形。不难发现，a1<0时这个推理也成立（劳动力同样需要ai个单位）。

代码

import sys
n=int(input()) #n个村庄
A=list(map(int,input().split())) #需求数组
if len(A)<=1:
    print(0)
res=0
for i in range(len(A)-1):
    res+=abs(A[i])
    A[i+1]+=A[i] # 把后面一个村庄的需求更新为本身需求加上前一个村庄的需求
print(res)

---

作者：AndrewYB
来源：CSDN
原文：https://blog.csdn.net/weixin_38391092/article/details/79585943