题意:给你一个数列,求出
求出所有的d(ai, aj) 的和 i和j 满足 1 <= i <= j <= n。 d(ai, aj) over all pairs (i, j) such that 1 ≤ i ≤ j ≤ n.
也就是说,每个数只能向后匹配,不能像前匹配。
一个数,它前面有m个数,后面有n 个数,它是会用n + m次的。只不过情况有两种。
我们利用map,可以很快找到有那些对于当前数特殊的数,也就是满足第二种情况的。两个数之间差为一,才会产生为零的情况(两个数相等,第一种也可以的)。
所以如果多加,那么减个一,少加,加个一。
#include<iostream>
#include<cstdio>
#include<algorithm>
#include<map>
#include<iomanip>
#include<cstdlib>
using namespace std;
typedef long long LL;
typedef long double db;//long double 精度高于18能满足本题要求
map<int, int> mp;
int main()
{
int n;
while(scanf("%d", &n) == 1)
{
mp.clear();
LL sum = 0, ans = 0;
for(int i = 0; i < n; i++)
{
int t;
scanf("%d", &t);
ans += 1LL * t * i;//注意这里可能会超int
ans -= sum;
sum += t;
if(mp.count(t + 1))
{
ans += mp[t + 1];
}
if(mp.count(t - 1))
{
ans -= mp[t - 1];
}
mp[t]++;
}
cout << fixed << setprecision(0) << ans << '\n';
}
return 0;
}