Codeforces 903 D Almost Difference

题目链接
因为写博客的时候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;
}
  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值