Spreading the Wealth UVA

Spreading the Wealth

 UVA - 11300 

题意：n个人坐一圈，每个人x[i]个糖果，且只能给相邻的人糖果，问最少一共给多少个糖果才能使得每个人糖果都一样多。

不看白书真是想不到这么做~

分析见白书P6

#include <bits/stdc++.h>
using namespace std;
#define ll long long
const int maxn=1000010;
ll c[maxn];
ll x[maxn];

int main(){
    int n;
    while(scanf("%d",&n)!=EOF){
        ll tot=0;
        for(int i=0;i<n;i++){
            scanf("%lld",&x[i]);
            tot+=x[i];
        }
        ll m=tot/n;
        c[0]=0;
        for(int i=1;i<n;i++) {
           c[i]=c[i-1]+x[i]-m;
        }
        sort(c,c+n);
        int base=c[n/2];
        ll res=0;
        for(int i=0;i<n;i++) res+=abs(base-c[i]);
        printf("%lld
",res);
    }
}