数据范围 0~1,000,000,000,int会溢出,必然要用__int64
最多10000组数据,暴力的话O(n^2)会超时
关键在排序后计算长度和,按照排序后的序列把相邻两个数看作一段,
如:1 2 3 4 5 (这里代表的是点的序号)
以 2-3 这一段为例, 1 ( 2 - 3 ) 4 5
计算2-3这一段在总的距离中出现的次数,因为只有跨越2-3这一段的长度才包含这一段,从1到3、4、5,2到3、4、5都包含这一段,可以得出这一段出现的次数为该段左侧的点数乘右侧的点数,以上计算的只是从左侧端点到右侧端点的次数,由于是双向的,反过来的结果是等效的,所以最后还要乘2
设当前一段的端点为a[i]和a[i+1],则这一段出现的次数表达式为:i * ( n - i ) * 2
题目中要求的是距离的总和,把每一段出现的次数乘该段的长度再求和即可
S = SUM( ( a[i+1] - a[i] ) * i * ( n - i ) * 2 ) ( i = 1, 2, ..., n )
Code:
#include <iostream>
#include <cstdio>
#include <algorithm>
using namespace std;
const int maxn = 10000 + 10;
__int64 num[maxn];
int main()
{
int n;
while( ~scanf( "%d", &n ) )
{
for( int i = 0; i < n; i++ )
cin >> num[i];
sort( num, num + n );
__int64 ans = 0;
for( int i = 1; i < n; i++ )
{
ans += ( num[i] - num[i-1] ) * i * ( n - i ) * 2;
}
cout << ans << endl;
}
return 0;
}