codeforces 903D Almost Difference

CF903D Almost Difference

题目描述

让我们来表示一个函数 给您一个数组,一个由n整数组成的数组.你必须计算所有对（i,j）的d值（a[i]，a[j]），比如1<=i<=j<=n。(差绝对值为1值为0，大于1为a[j]-a[i])

输入数据

第一行包含一个整数n（1<=n<=200000）——a中的元素数。 第二行包含n整数a[1]、a[2]、…、a[n]（1<=a[i]<=10^9）——数组的元素。

输出数据

在所有对（i,j）中打印一个整数，即d（a[i]，a[j]）的总和，例如1<=i<=j<=n。

思路：前缀和+思维

利用前缀和+遍历得出a[j]-a[i]的和，再通过在线计数的方式，若值相等不影响，若当前值-1出现一次，则需将多加的次数减去，同理+1一样。注意本题一个坑点是long long范围不够，需用long double

//mode by lsy
#include<bits/stdc++.h>
#define ll long long
#define pq priority_queue
#define repr(i,x,y) for(int i=x;i<=y;i++)
#define rep(i,x,y) for(int i=x;i<y;i++)
#define drep(i,x,y) for(int i=x;i>=y;i--)
#define pb push_back
#define P make_pair
#define pii pair<int,int>
#define mem(x,y) memset(x,y,sizeof(x))
using namespace std;
const int inf = 0x3f3f3f3f, maxn = 2e5 + 5;
const double eps = 1e-8;
int n;
ll a[200005];
ll sum[200005];
int main() {
    cin>>n;
    repr(i,1,n){
        cin>>a[i];
        sum[i]=sum[i-1]+a[i];//前缀和
    }
    long double ans=0;
    map<ll,ll> cnt;//计数
    cnt.clear();
    repr(i,1,n){
        ans+=(long double)(i-1)*a[i]-sum[i-1];//Σ(a[i]-a[temp])(temp<i)
        ans-=cnt[a[i]-1];//当前a[i]-1出现次数
        ans+=cnt[a[i]+1];
        cnt[a[i]]++;
    }
    cout<<fixed<<setprecision(0)<<ans<<'\n';
    //cout<<endl<<clock();
    return 0;
}
/*
    Simplicity is a prerequisite for stability.
            ——by acmer lsy
*/