CF903 D.Almost Difference

题目：http://codeforces.com/contest/903/problem/D
分析：
一开始做的好麻烦（离散化+前缀和+map），后来看了别人的代码简化了一下（前缀和+map）。
先说下思路，对第i个数，不考虑距离在1之内的情况，对答案的贡献是(i-1)*a[i]-sum，sum是1..i-1的前缀和；
注意： 要用longdouble！若输出用%.0Lf，需要用c++14，当然也可以用 cout << fixed << setprecision(0) << ans 这个输出。
代码：
离散化+前缀和+map

#include <bits/stdc++.h>using namespace std;const int Tmax=200005;int n,p;long double sum[Tmax],data[Tmax],what[Tmax],tmp[Tmax],num[Tmax],ans;map<double,int> lisan;bool equal(double x,double y){    return fabs(x-y)<=0.0000001;}void work(){    int i;    num[lisan[data[1]]]++;    for(i=2;i<=n;i++)    {        ans+=data[i]*(i-1)-sum[i-1];        if(equal(what[lisan[data[i]]-1],data[i]-1)==true)//if(what[lisan[data[i]]-1]==data[i]-1)        {            ans-=1.0*num[lisan[data[i]]-1];        }        if(equal(what[lisan[data[i]]+1],data[i]+1)==true)//if(what[lisan[data[i]]+1]==data[i]+1)        {            ans+=1.0*num[lisan[data[i]]+1];        }        num[lisan[data[i]]]+=1.0;        //printf("%d ",ans);    }    //printf("\n");    return;}int main(){    int i;    scanf("%d",&n);    for(i=1;i<=n;i++)    {        scanf("%Lf",&data[i]);        tmp[i]=data[i];    }    sort(tmp+1,tmp+1+n);    tmp[0]=-10;    for(i=1;i<=n;i++)    {        if(equal(tmp[i],tmp[i-1])==false)        {            lisan[tmp[i]]=++p;            what[p]=tmp[i];        }    }    //for(i=1;i<=n;i++) printf("%d ",what[i]);    ///printf("\n");    for(i=1;i<=n;i++)    {        sum[i]=sum[i-1]+data[i];        //printf("%d ",sum[i]);    }    //printf("\n");    work();    cout<<fixed<<setprecision(0)<<ans;//printf("%.0Lf",ans);    return 0;}

前缀和+map

#include <bits/stdc++.h>using namespace std;const int Tmax=200005;int n,p;long long int sum,tmp;map<long long int,int> MM;long double ans;int main(){    int i;    scanf("%d",&n);    for(i=1;i<=n;i++)    {        scanf("%I64d",&tmp);        ans+=(i-1)*tmp-sum+MM[tmp+1]-MM[tmp-1];        sum+=tmp;        MM[tmp]++;    }    cout << fixed << setprecision(0) << ans;    //printf("%.0Lf",ans);    return 0;}