BZOJ 3781: 小B的询问-优快云博客

本文链接：https://blog.youkuaiyun.com/bababaab/article/details/52103693

本文详细介绍了一种基于莫队算法的实现方式，通过具体示例BZOJ2038来展示如何优化区间查询问题。该算法适用于区间操作与查询，通过离线预处理提高效率。

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

裸奔一个莫队
和这个BZOJ 2038: [2009国家集训队]小Z的袜子(hose)很像

#include<cmath>
#include<cstdio>
#include<cstring>
#include<algorithm>
#define g getchar()
#define ll long long
#define inf 0x3f3f3f3f
using namespace std;
inline ll read(){
    ll x=0,f=1;char ch=g;
    for(;ch<'0'||ch>'9';ch=g)if(ch=='-')f=-1;
    for(;ch>='0'&&ch<='9';ch=g)x=x*10+ch-'0';
    return x*f;
}
inline void out(ll x){
    int a[25],wei=0;
    if(x<0)putchar('-'),x=-x;
    for(;x;x/=10)a[++wei]=x%10;
    if(wei==0){puts("0");return;}
    for(int j=wei;j>=1;--j)putchar('0'+a[j]);
    putchar('\n');
}
struct re{int l,r,id;}ask[50005];
ll now,ans[50005];
int num[50005],a[50005],block,n,Q,k;
inline int update(int p,int x){
    now-=(ll)num[a[p]]*(ll)num[a[p]];
    num[a[p]]+=x;
    now+=(ll)num[a[p]]*(ll)num[a[p]];
}
inline bool cmp(re x,re y){return x.l/block==y.l/block?x.r<y.r:x.l<y.l;}

int main(){
//  freopen("","r",stdin);
//  freopen("","w",stdout);
    n=read(),Q=read(),k=read();
    for(int i=1;i<=n;++i)
    a[i]=read();
    for(int i=1;i<=Q;++i)
    ask[i].l=read(),ask[i].r=read(),ask[i].id=i;
    block=(int)sqrt(n);
    sort(ask+1,ask+1+Q,cmp);
    now=1;num[a[1]]=1;
    for(int i=1,r=1,l=1;i<=Q;++i){
        for(;r<ask[i].r;)++r,update(r,1);
        for(;l>ask[i].l;)--l,update(l,1);
        for(;r>ask[i].r;)update(r,-1),--r;
        for(;l<ask[i].l;)update(l,-1),++l;
        ans[ask[i].id]=now;
    }
    for(int i=1;i<=Q;++i)out(ans[i]);
    return 0;
}