分块/莫队算法Codeforces86D

最新推荐文章于 2023-08-07 09:20:49 发布

原创最新推荐文章于 2023-08-07 09:20:49 发布 · 281 阅读

0 ·

CC 4.0 BY-SA版权

分块同时被 2 个专栏收录

5 篇文章

订阅专栏

莫队算法

5 篇文章

订阅专栏

本文提供了一道CodeForces平台上的难题D的详细解答，采用分块算法优化查询过程，通过预处理和动态更新策略减少时间复杂度，适用于算法竞赛及数据结构学习。

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

题目链接：http://codeforces.com/contest/86/problem/D

#include <bits/stdc++.h>

using namespace std;
using ll = long long ;
const int N = 1E6 + 7;
int a[N], belong[N], BLOCK, sum[N];
ll res, ans[N];
struct node
{
    int l, r, id;
    bool operator < (const node & x) const {
        if(belong[l] == belong[x.l]) return r < x.r;
        else return l < x.l;
    }
}Q[N];
void modify(int p, int add)
{
    res -= a[p] * 1ll * sum[a[p]] * sum[a[p]];
    sum[a[p]] += add;
    res += a[p] * 1ll * sum[a[p]] * sum[a[p]];
}
int main()
{
    int n, t;
    scanf("%d%d",&n,&t);
    BLOCK = sqrt(n);
    for(int i = 1;i <= n;i ++) scanf("%d",&a[i]), belong[i] = (i - 1) / BLOCK + 1;
    for(int i = 1;i <= t;i ++) {
        scanf("%d%d",&Q[i].l,&Q[i].r);
        Q[i].id = i;
    }
    sort(Q+1, Q+1+t);
    for(int i = 1, l = 1, r = 0;i <= t;i ++) {
        for(;r < Q[i].r; r ++) modify(r+1, 1);
        for(;r > Q[i].r; r --) modify(r, -1);
        for(;l < Q[i].l; l ++) modify(l, -1);
        for(;l > Q[i].l; l --) modify(l-1, 1);
        ans[Q[i].id] = res;
    }
    for(int i = 1; i <= t; i ++) printf("%lld\n",ans[i]);
    return 0;
}