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
*/