CSU 1515 Sequence

莫队算法+map

#include<cstdio>
#include<cstring>
#include<cmath>
#include<map>
#include<algorithm>
using namespace std;

const int maxn=+;
int n,t,a[maxn],cnt[maxn*],pos[maxn];
struct X
{
    int l,r,id;
} s[maxn];
int L,R;
int Ans,f[maxn];

bool cmp(const X&a,const X&b)
{
    if(pos[a.l]==pos[b.l]) return a.r<b.r;
    return pos[a.l]<pos[b.l];
}

int main()
{
    while(~scanf("%d",&n))
    {
        scanf("%d",&t);
        int sz=sqrt(n);
        for(int i=; i<=n; i++)
        {
            scanf("%d",&a[i]);
            pos[i]=i/sz;
        }
        for(int i=; i<=t; i++)
        {
            scanf("%d%d",&s[i].l,&s[i].r);
            s[i].id=i;
        }

        sort(s+,s++t,cmp);

        map<int,int>m;
        Ans=;
        for(int i=s[].l; i<=s[].r; i++)
        {
            Ans=Ans+m[a[i]-]+m[a[i]+];
            m[a[i]]++;
        }

        f[s[].id]=Ans;
        L=s[].l;
        R=s[].r;

        for(int i=; i<=t; i++)
        {
            while(L<s[i].l)
            {
                Ans=Ans-m[a[L]-]-m[a[L]+];
                m[a[L]]--;
                L++;
            }

            while(L>s[i].l)
            {
                L--;
                Ans=Ans+m[a[L]-]+m[a[L]+];
                m[a[L]]++;
            }

            while(R>s[i].r)
            {
                Ans=Ans-m[a[R]-]-m[a[R]+];
                m[a[R]]--;
                R--;
            }

            while(R<s[i].r)
            {
                R++;
                Ans=Ans+m[a[R]-]+m[a[R]+];
                m[a[R]]++;
            }

            f[s[i].id]=Ans;
        }

        for(int i=; i<=t; i++)
            printf("%d\n",f[i]);
    }

    return ;
}