NBUT - 1457 莫队算法

Sona

给你一个序列，问你(L,R)区间内每个数出现次数的立方和（开始当成了平方和。。。）莫队模板题，需要离散化一下。

#include<iostream>
#include<cstdio>
#include<cmath>
#include<cstring>
#include<algorithm>

using namespace std;
typedef long long ll;
const int maxn=100050;
int a[maxn],t,b[maxn];
ll c[maxn],an[maxn];
struct node{
    int l,r,id;
}s[maxn];
bool cmp(const node &a,const node &b)
{
    if(a.l/t==b.l/t) return a.r>b.r;
    return a.l<b.l;

}
int n,m,N;

int main()
{
    //freopen("in.txt","r",stdin);
    while(~scanf("%d",&n))
    {
        for(int i=1;i<=n;i++)
        {
            scanf("%d",&a[i]);
            b[i]=a[i];
        }
        sort(b+1,b+n+1);
        N=unique(b+1,b+n+1)-b-1;
        for(int i=1;i<=n;i++)
        {
            a[i]=lower_bound(b+1,b+N+1,a[i])-b-1;
        }
        for(int i=0;i<=N+1;i++)
            c[i]=0;
        scanf("%d",&m);
        for(int i=1;i<=m;i++)
        {
            scanf("%d%d",&s[i].l,&s[i].r);
            s[i].id=i;
        }
        t=(int)(sqrt(1.0*n)+0.5);///这里写sqrt（n）会编译错误
        sort(s+1,s+m+1,cmp);

        int l=s[0].l,r=s[0].l-1;
        ll ans=0;
        //memset(c,0,sizeof c);
        for(int i=1;i<=m;i++)
        {
            while(l<s[i].l)
            {
                ans-=c[a[l]]*c[a[l]]*c[a[l]];
                c[a[l]]--;
                ans+=c[a[l]]*c[a[l]]*c[a[l]];
                l++;
            }

            while(l>s[i].l)
            {
                l--;
                ans-=c[a[l]]*c[a[l]]*c[a[l]];
                c[a[l]]++;
                ans+=c[a[l]]*c[a[l]]*c[a[l]];
            }

            while(r<s[i].r)
            {
                r++;
                ans-=c[a[r]]*c[a[r]]*c[a[r]];
                c[a[r]]++;
                ans+=c[a[r]]*c[a[r]]*c[a[r]];
            }

            while(r>s[i].r)
            {
                ans-=c[a[r]]*c[a[r]]*c[a[r]];
                c[a[r]]--;
                ans+=c[a[r]]*c[a[r]]*c[a[r]];
                r--;
            }
            an[s[i].id]=ans;
        }
        for(int i=1;i<=m;i++)
            printf("%I64d\n",an[i]);
    }
    return 0;
}