bzoj3781: 小B的询问莫队算法

最新推荐文章于 2018-10-31 18:32:36 发布

原创最新推荐文章于 2018-10-31 18:32:36 发布 · 576 阅读

0 ·

CC 4.0 BY-SA版权

bzoj 同时被 2 个专栏收录

88 篇文章

订阅专栏

乱搞

27 篇文章

订阅专栏

本文深入解析了裸莫队算法的原理及其在解决区间更新和查询问题上的高效实现。通过实例演示了如何利用裸莫队算法优化复杂度，显著提升算法性能。

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

裸莫队，没什么好说的。。。。。。。

#include <iostream>
#include <algorithm>
#include <cstdio>
#include <cmath>
#include <cstring>
using namespace std;
#define maxn 60000
typedef long long sint;
int n,m,k;
int save[maxn],pos[maxn];
sint ans,s[maxn];
struct node
{
    int l,r,id;
    sint ans;
}a[maxn];
sint gcd(sint a,sint b)
{
    if(!b) return a;
    return gcd(b,a%b);
}
sint sqr(sint x)
{
    return x*x;
}
bool cmp(node aa,node bb)
{
    if(pos[aa.l]==pos[bb.l]) return aa.r<bb.r;
    return aa.l<bb.l;
}
bool cmp2(node aa,node bb)
{
    return aa.id<bb.id;
}
void update(int p,int add)
{
	ans-=sqr(s[save[p]]);
	s[save[p]]+=add;
	ans+=sqr(s[save[p]]);
}
void solve()
{
    for(int i=1,l=1,r=0;i<=m;i++)
    {
        for(;r>a[i].r;r--)
            update(r,-1);
        for(;r<a[i].r;r++)
            update(r+1,1);
        for(;l<a[i].l;l++)
            update(l,-1);
        for(;l>a[i].l;l--)
            update(l-1,1);
		a[i].ans=ans;
    }
}
int main()
{
    scanf("%d%d%d",&n,&m,&k);
    for(int i=1;i<=n;i++) scanf("%d",&save[i]);
    int len=sqrt(n);
    for(int i=1;i<=n;i++) pos[i]=(i-1)/len+1;
    for(int i=1;i<=m;i++)
    {
        scanf("%d%d",&a[i].l,&a[i].r);
        a[i].id=i;
    }
    sort(a+1,a+1+m,cmp);
    solve();
    sort(a+1,a+1+m,cmp2);
    for(int i=1;i<=m;i++)
    {
        printf("%lld\n",a[i].ans);
    }
    return 0;
}