贪心,本来想的贪心方法是找到每一个需求量为正数的居民,往两边搜,搜到最近的那个需求量为负的居民,权衡需求量与销售量不难最少使其中的一个变为零,另外一个如果不为零,则减去或加上另一个的需求量,不断重复操作,直到所有居民的数值均为零。虽然后来中间也做了不少剪枝,但最后还是TLE了。
后来参考了http://hi.baidu.com/knowledgetime/blog/item/fbcfe9c9b6ad4048f21fe7d1.html的贪心方法,觉得这个方法非常巧妙,可以将其看成每次只有邻居间互相买卖葡萄酒,最后统计总的出入金额便可。
代码如下:
#include<iostream> #include<cstdlib> #include<cstring> #include<cstdio> #include<cmath> using namespace std; int a[100002]; int main() { #ifdef test freopen("in.txt", "r", stdin); #endif int num; while(scanf("%d", &num), num) { long long sum = 0, fm = 0; for(int i = 0; i < num; i++) scanf("%d", &a[i]); for(int i = 0; i < num; i++) { sum += abs(fm); fm += a[i]; } printf("%lld\n", sum); } return 0; }