Codeforces 86D. Powerful array

最新推荐文章于 2023-03-29 23:41:56 发布

原创最新推荐文章于 2023-03-29 23:41:56 发布 · 1k 阅读

0 ·

CC 4.0 BY-SA版权

文章标签：

#codeforces #莫队算法

codeforces 同时被 2 个专栏收录

4 篇文章

订阅专栏

莫队算法

2 篇文章

订阅专栏

本文介绍了一个使用莫队算法解决特定区间问题的例子。该问题要求计算区间[L,R]内所有Ks*Ks*s的和，通过巧妙地运用莫队算法进行区间查询优化，实现了高效的求解。

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

题意：Ks为区间内s的数目，求区间[L,R]之间所有Ks*Ks*s的和

分析：用莫队算法可以做，时间卡得有点紧，写的时候注意

#include <iostream>
#include<cstdio>
#include<algorithm>
#include<string>
#include<cstring>
#define LL long long
#define EPS 1e-9
#include<cmath>
#define N 200005
#define M 1000005
using namespace std;
LL ans[N],tmp;
int a[N];
LL num[M];
struct query{
    int l,r,id,pos;
    bool operator <(const query &k)const{
        return pos!=k.pos?pos<k.pos:r<k.r;
    }
}q[N];
void insert(int x){
    tmp+=((num[a[x]]<<1)+1)*a[x]; //这里容易超时
    num[a[x]]++;
}
void erase(int x){

    num[a[x]]--;
    tmp-=((num[a[x]]<<1)+1)*a[x];
}
int main()
{
    int n,t,i,l,r,unit;
    while(~scanf("%d%d",&n,&t)){
        memset(num,0,sizeof(num));
        for(i=1;i<=n;i++){
            scanf("%d",&a[i]);
        }
        unit=sqrt((double)n);
        for(i=0;i<t;i++){
            scanf("%d%d",&q[i].l,&q[i].r);
            q[i].id=i;
            q[i].pos=q[i].l/unit;
        }
        sort(q,q+t);
        l=1;
        r=0;
        tmp=0;
        for(i=0;i<t;i++){
            while(r<q[i].r)insert(++r);
            while(r>q[i].r)erase(r--);
            while(l<q[i].l)erase(l++);
            while(l>q[i].l)insert(--l);
            ans[q[i].id]=tmp;
        }
        for(i=0;i<t;i++){
            printf("%I64d\n",ans[i]);
        }
    }
    return 0;
}