codeforces 903D Almost Difference(递推式)

题意很简单，自己读吧

思路：
对于 1,2,3,1,3，记录前缀和，思考前面的所有数到这个数的情况
a[i]*(i-1) - sum[i-1]，即不考虑(a[i]-1和a[i]+1的数时，前面所有数到a[i]的结果，因为对于每个a[i]-1，相当于多加了一个1，对于每个a[i]+1，相当于多减了个1，所以用map统计前面每个数出现的次数。最后再把距离为1的数补回来即可。

代码如下：

#include<iostream>#include<cstdio>#include<cmath>#include<cstring>#include<vector>#include<string>#include<algorithm>#include<map>#include<set>#include<vector>using namespace std;typedef long double ld;typedef long long ll;const int MAX = 200010;ll sum[MAX];ll a[MAX],N;map<ll,int> mp;int main(void){    memset(sum,0,sizeof(sum));    scanf("%lld",&N);    for(int i=1;i<=N;++i){        scanf("%lld",&a[i]);        sum[i] = sum[i-1] + a[i];    }    ld res = 0;//最大的可能结果是N^N/2*a[i]所以肯可能爆ll,这题用ld。    for(int i=1;i<=N;++i){        ld temp = (ld)a[i];//考虑对于这个一位数的情况.        temp = temp*(i-1);        temp -= sum[i-1];        if(mp.count(a[i] - 1)){//去掉多加的            temp -= mp[a[i]-1];        }        if(mp.count(a[i]+1)){//补上多减的            temp += mp[a[i]+1];        }        mp[a[i]]++;        res += temp;    }    printf("%.0Lf\n",res);    return 0;}