莫队算法:洛谷:2709

莫队入门板题:

对于离线操作可以利用莫队来写,利用分块思想,进行排序,将区间距离最小化,实现O(nsqrt(n))的时间复杂度,对于贡献值,如果区间需要就加上,如果区间不需要就减去

题目链接:小B的询问

AC代码:

#include<bits/stdc++.h>
using namespace std;
//莫队算法
const int maxn=1e5+5;
int pos[maxn],a[maxn],c[maxn];
long long ans[maxn],res;
struct node
{
    int l,r,poi;
}q[maxn];
bool cmp(node x,node y)//排序可以减少复杂度
{
    if(pos[x.l]==pos[y.l])
    {
        return x.r<y.r;
    }
    return pos[x.l]<pos[y.l];
}
void add(int i)
{
    c[a[i]]++;
    res+=c[a[i]]*c[a[i]]-(c[a[i]]-1)*(c[a[i]]-1);
}
void sub(int i)
{
    c[a[i]]--;
    res-=(c[a[i]]+1)*(c[a[i]]+1)-c[a[i]]*c[a[i]];
}
int main()
{
    int n,m,k;
    scanf("%d %d %d",&n,&m,&k);
    int dis=sqrt(n);
    for(int i=1;i<=n;i++)
    {
        scanf("%d",&a[i]);
        pos[i]=i/dis;
    }
    for(int i=0;i<m;i++)
    {
        scanf("%d %d",&q[i].l,&q[i].r);
        q[i].poi=i;
    }
    sort(q,q+m,cmp);
    int l=1,r=0;
    for(int  i=0;i<m;i++)
    {
       while(l<q[i].l)
       {
           sub(l++);
       }
       while(r>q[i].r)
       {
           sub(r--);
       }
       while(l>q[i].l)
       {
           add(--l);
       }
       while(r<q[i].r)
       {
           add(++r);
       }
       ans[q[i].poi]=res;
    }
    for(int i=0;i<m;i++)
    {
        printf("%lld\n",ans[i]);
    }
}
发布了205 篇原创文章 · 获赞 12 · 访问量 8505

猜你喜欢

转载自blog.csdn.net/yangzijiangac/article/details/103904914
今日推荐