【t041】距离之和

Time Limit: 1 second
Memory Limit: 128 MB

【问题描述】

在一条数轴上有N头牛在不同的位置上，每头牛都计算到其它各头牛的距离。求这n*(n-1)个距离的总和。
【数据规模】
1<= N <= 10000。每头牛所在位置是一个范围在0到1,000,000,000之内的整数。

【输入格式】

第一行：N
后面N行，每行一个整数，表示一头牛所在位置。
【输出格式】

一个整数。
测试点说明：
说明：
（1+2+3+4)+(4+3+2+1)
+(2+1+1+2)+(1+1+2+3)
+(3+2+1+1) = 40
Sample Input

5
1
5
3
2
4

Sample Output

40

【题目链接】:http://noi.qz5z.com/viewtask.asp?id=t041

【题解】

将所有的牛的位置按照升序排;
设dis[i]表示第i头牛到前i-1头牛的总距离;
则dis[i] = dis[i-1]+(i-2)*(a[i]-a[i-1])+a[i]-a[i-1];
其中dis[i-1]+(i-2)*(a[i]-a[i-1])表示i和前i-2头牛各连i-2条边的距离和(以i-1号牛为中转站,增加i-2条a[i]-a[i-1]边权的边就能表示这个状态了),然后再加上一条a[i]-a[i-1]边，表示在第i号牛和第i-1号牛之间再连一条边;
最后∑dis[i];
然后再乘2；(题目要求的是A(n,2)的排列);

【完整代码】

#include <iostream>
#include <cstdio>
#include <algorithm>
#define LL long long

using namespace std;

const int MAXN = 1e4+10;

int n;
LL a[MAXN];
LL dis[MAXN];
LL ans = 0;

int main()
{
    //freopen("F:\rush.txt","r",stdin);
    cin >> n;
    for (int i = 1;i <= n;i++)
        cin >> a[i];
    sort(a+1,a+1+n);
    dis[1] = 0;dis[2] = a[2]-a[1];
    for (int i = 3;i <= n;i++)
    {
        dis[i] = dis[i-1]+(i-2)*(a[i]-a[i-1]);
        dis[i]+=a[i]-a[i-1];
    }
    for (int i = 1;i <= n;i++)
        ans+=dis[i];
    cout << ans*2<<endl;
    return 0;
}