一道阿里巴巴笔试题

有一个淘宝商户，在某城市有n个仓库，每个仓库的储货量不同，现在要通过货物运输，将每次仓库的储货量变成一致的，n个仓库之间的运输线路围城一个圈，即1->2->3->4->...->n->1->...，货物只能通过连接的仓库运输，设计最小的运送成本（运货量*路程）达到淘宝商户的要求，并写出代码。

解答：这个题目类似的题目有：

题目：http://www.lydsy.com/JudgeOnline/problem.php?id=1045
有n个小朋友坐成一圈，每人有ai个糖果。每人只能给左右两人传递糖果。每人每次传
递一个糖果代价为1，求使所有人获得均等糖果的最小代价。
分析：
假设a1分给an的糖果数为k，则可以得到以下的信息：
  a1              a2                       　a3　　　　　　　　   an-1 　　　　　　　　　　　　 an
当前数目：a1-k           a2　　　　　　　　 a3　　　　　　　　   an-1 　　　　　　　　　　　　 an+k
所需代价：|a1-k-ave| |a1+a2-k-2*ave| |a1+a2+a3-k-3*ave||a1+..+a(n-1)-k-(n-1)*ave|   |k|
以sum[i]表示从a1加到ai减掉i*ave的和值，这以上可以化简为
总代价 = |s1-k|+|s2-k|+...+|s(n-1)-k|+|k|
不难看出：当k为s1...s(n-1)中的中位数的时候，所需的代价最小

#include <cstring>
#include <iostream>
#include <algorithm>
                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                                          
using namespace std;
const int X = 1000005;
typedef long long ll;
ll sum[X],a[X];
ll n;
ll Abs(ll x){
    return max(x,-x);
}
int main(){
    while(cin>>n){
        ll x;
        ll tot = 0;
        for(int i=1;i<=n;i++){
            scanf("%lld",&a[i]);
            tot += a[i];
        }
        ll ave = tot/n;
        for(int i=1;i<n;i++)
            sum[i] = a[i]+sum[i-1]-ave;
        sort(sum+1,sum+n);
        ll mid = sum[n/2];
        ll ans = Abs(mid);
        for(int i=1;i<n;i++)
            ans += Abs(sum[i]-mid);
        cout<<ans<<endl;
    }
    return 0;
}