Bzoj3236: [Ahoi2013]作业

Bzoj3236: [Ahoi2013]作业

刚学完莫队，不过这题用莫队的复杂度算起来好高啊，居然能过……

像求逆序对一样用两个树状数组维护区间中每个数值出现的个数，以及每个数是否出现，前者用来求答案1，后者用来求答案2。

在区间伸缩时加入删除最后更新答案即可。

这次压行压的有点厉害，和DeepinC有的一拼了……

 

#include<algorithm>
#include<iostream>
#include<cstdio>
#include<cmath>
using namespace std;
int L[1010],R[1010],pos[100010];
struct ques
{
    int L,R,a,b,id;
    #define L(x)  que[x].L
    #define R(x)  que[x].R
    #define a(x)  que[x].a
    #define b(x)  que[x].b
    #define id(x) que[x].id
    friend bool operator < (ques a,ques b)    
    {
        return (pos[a.L]^pos[b.L])?(pos[a.L]<pos[b.L]):((pos[a.L]&1)?(a.R<b.R):(a.R>b.R));
    }
}que[1000010];
int C[100010],C2[100010],cnt[100010];
int n,m,a[100010];
int eans1[1000010],eans2[1000010];
#define lowbit(x) ((x)&(-(x)))
inline int read()
{
    int s=0;char a=getchar();
    while(a<'0'||a>'9')a=getchar();
    while(a>='0'&&a<='9'){s=s*10+a-'0';a=getchar();}
    return s;
}
void add(int x,int y)
{while(x<=100010){C[x]+=y;x+=lowbit(x);}}
int ask(int x)
{int ans=0;while(x){ans+=C[x];x-=lowbit(x);}return ans;}
void add2(int x,int y)
{while(x<=100010){C2[x]+=y;x+=lowbit(x);}}
int ask2(int x)
{int ans=0;while(x){ans+=C2[x];x-=lowbit(x);}return ans;}
signed main()
{
    cin>>n>>m;
    for(int i=1;i<=n;i++)a[i]=read();
    for(int i=1;i<=m;i++)
    L(i)=read(),R(i)=read(),a(i)=read(),b(i)=read(),id(i)=i;
    int t=sqrt(n);    
    for(int i=1;i<=t;i++)
        L[i]=R[i-1]+1,R[i]=i*t;
    if(R[t]<n)t++,L[t]=R[t-1]+1;R[t]=n;
    for(int i=1;i<=t;i++)
    for(int j=L[i];j<=R[i];j++)
        pos[j]=i;
    sort(que+1,que+m+1);
    int l=1,r=0,ans1=0,ans2=0;
    for(int i=1;i<=m;i++)
    {
        ans1=ask(b(i))-ask(a(i)-1);
        while(l>L(i)){l--,add(a[l],1),ans1=ask(b(i))-ask(a(i)-1);if(cnt[a[l]]==0)add2(a[l],1);cnt[a[l]]++;  }
        while(l<L(i)){add(a[l],-1),ans1=ask(b(i))-ask(a(i)-1);if(cnt[a[l]]==1)add2(a[l],-1);cnt[a[l]]--,l++;}
        while(r<R(i)){r++,add(a[r],1),ans1=ask(b(i))-ask(a(i)-1);if(cnt[a[r]]==0)add2(a[r],1);cnt[a[r]]++;  }
        while(r>R(i)){add(a[r],-1),ans1=ask(b(i))-ask(a(i)-1);if(cnt[a[r]]==1)add2(a[r],-1);cnt[a[r]]--,r--;}
        ans2=ask2(b(i))-ask2(a(i)-1);
        eans1[id(i)]=ans1,eans2[id(i)]=ans2;
    }
    for(int i=1;i<=m;i++)
        printf("%d %d\n",eans1[i],eans2[i]);
}