9.27周赛丨cf903D（前缀和+思维）

题目链接：https://codeforces.com/problemset/problem/903/D

知识点：前缀和+思维

这个题会了以后，会发现他是一个很简单的题。刚开始做的时候，直接考虑的二重循环，很显然他过不去。甚至于写了离散化，但是就是无用功而已。于是隐隐约约觉得会用到前缀和，但是也不清楚具体要怎么用。

本题要保证逐项相减的顺序，保证是后面的减前面的，同时考虑约束条件。

当我们用当前的数减去之前的数的时候，很容易得到一个式子：a[i]*(i-1)-sum[i-1]。再对于a[i],当i前面的数出现a[i]-1和a[i]+1进行相应的加减操作。

 

• 本题出现了map，一个到现在非常不熟的点。
• 不能用long long，要用long double，实在没想到还有这个啊。
• 那个定义格式的方式，我已经完全忘记了。

#include<iostream>
#include<cstdio>
#include<map>
#include<iomanip>
typedef long long ll;
typedef long double ld;
ld a[202000];
ld sum[202000];
using namespace std;
int main(){
	int n;
	scanf("%d",&n);
	
	for(int i= 1;i<=n;i++){
		cin>>a[i];
		sum[i] = sum[i-1] + a[i] ;
	}
	
	ld ans = 0;
	map<ld,ld>cnt;
	
	for(int i = 1;i<=n;i++){
		ans += a[i] * (i - 1) -sum[i-1];
		ans -= cnt[a[i]-1];
		ans += cnt[a[i]+1];
		cnt[a[i]]++;
	}
	
	cout<<fixed<<setprecision(0)<<ans;
	return 0;
}