题目链接
因为写博客的时候cf打不开,所以就暂时通过这个虚拟OJ来做题了QWQ
题目分析
题目的意思大概就是说从数列中任意取出两个数,如果这两个元素的差值大于1,那我们就把这个差值记下来,如果差值小于等于1,就不计算。最后求出那些差值大于1的差值的总和。
初次看题目时还以为是一道DP,后面想了一下,应该不是DP题。在想了想,发现好像要用到前缀和,不然不可能在O(N)的时间内过掉这道题。不过后面再想也就不知道该怎么想下去了。于是翻了一下网上的题解,看了一会儿才懂。在这里解释一下吧。
我们可以用这样一个式子来表示答案:x*(i-1)-sum[i-1]。这个式子相当于是表示用i个x把x前面的每一个数都减去一次,这样做后我们可以发现有些数是多减了的,有些事少减了的。比如这样一个数列:4 4 1 2 3 1 3。我们用第5个数“3”乘以4减去(4+4+1+2)后作为答案,发现答案不是正确的,原本4,4这两个数是不对答案做出贡献的,然而我们却多减了1,所以现在要加回来,而原来的2也不对答案产生贡献,但在计算的时候我们少减了,所以要重新加回来1个1。我们在这里用map来记录一下前面的数字出现的次数,方便后面计算的时候加减,具体见程序代码。
#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<cmath>
#include<map>
#include<iomanip>
using namespace std;
int n;
long double ans;
long long sum,x;
map<long long,int>mp;
inline long long read(){
long long x=0,w=1;char ch=getchar();
while(ch!='-'&&(ch<'0'||ch>'9')) ch=getchar();
if(ch=='-') w=-1,ch=getchar();
while(ch>='0'&&ch<='9') x=(x<<3)+(x<<1)+(ch-48),ch=getchar();
return x*w;
}
int main(){
n=read();
for(register int i=1;i<=n;++i){
long long x=read();
ans+=(long double)x*(i-1)-(long double)sum;
//先不管会不会减多减少,全部都减掉
if(mp[x-1]) ans-=(long double)mp[x-1];
//如果比在x之前有比x小1的数字存在,就减去
if(mp[x+1]) ans+=(long double)mp[x+1];
//如果比在x之前有比x大1的数字存在,就减去
sum+=x;++mp[x];
}
cout << fixed << setprecision(0) << ans << endl;
return 0;
}