Codeforces 903D - Almost Difference

最新推荐文章于 2022-06-26 21:55:37 发布

TooSIMple_

最新推荐文章于 2022-06-26 21:55:37 发布

阅读量484

点赞数

分类专栏： CodeForces

本文链接：https://blog.youkuaiyun.com/i1020/article/details/78808056

版权

CodeForces 专栏收录该内容

23 篇文章

订阅专栏

本文提供CodeForces 903D题目的解决方案，介绍了一种使用前缀和及映射计数的方法来计算特定序列的距离之和，并通过C++代码实现，适用于算法竞赛和数据结构学习。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

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

思路：对于一个给定的a[x]来说，对于 d(a[1],a[x]) + d(a[2],a[x])+.......+d(a[i].a[x])的结果是 i * a[x] - sum(sum为前x-1项元素的和) + (a[i] - a[x] == 1的所有数的个数，即所有等于a[x]+1的数的个数) - (a[x] - a[i] == 1 的所有数的个数，即所有等于a[x] - 1 的数的个数)

setprecision(0) 和 fixed 控制输出的精度

#include <bits/stdc++.h>
#define LL long long
#define ld long  double
using namespace std;

int n;
LL a[210000];

int main(){
    scanf("%d",&n);
    map<LL, LL> mp;
    for(int i = 0; i < n; i ++) scanf("%lld",&a[i]);
    ld ans = 0, sum = 0;
    for(int i = 0; i < n; i ++){
        ans += i * a[i] - sum + mp[a[i] + 1] - mp[a[i] - 1];
        mp[a[i]] ++;
        sum += a[i];
    }
    cout << fixed << setprecision(0) << ans << endl;
    return 0;
}