BZOJ3236: [Ahoi2013]作业（权值分块+莫队）

传送门
听说一堆莫队+树状数组过的。。正解不应该是$O(n\sqrt{n})$？
莫队配合权值分块有奇效。。

#include<bits/stdc++.h>
using namespace std;
inline int read(){
    char ch=getchar();int i=0,f=1;
    while(!isdigit(ch)){if(ch=='-')f=-1;ch=getchar();}
    while(isdigit(ch)){i=(i<<1)+(i<<3)+ch-'0';ch=getchar();}
    return i*f;
}
inline void W(int x){
    static int buf[50];
    if(!x){putchar('0');return;}
    if(x<0){putchar('-');x=-x;}
    while(x){buf[++buf[0]]=x%10;x/=10;}
    while(buf[0])putchar(buf[buf[0]--]+'0');
}
const int Maxn=1e5+50,Maxm=1e6+50,Smax=550;
struct Q{
    int l,r,a,b,bl,id;
    friend inline bool operator <(const Q &a,const Q &b){
        return a.bl<b.bl||(a.bl==b.bl&&a.r<b.r);
    }
}qry[Maxm];
int n,m,a[Maxn],lsh[(Maxm<<1)+Maxn],tot,S_qry,S_val,ans1[Maxm],ans2[Maxm],bg[Maxn],ed[Maxn];
int bl_cnt[Smax],bl_ans[Smax],cnt[(Maxm<<1)+Maxn],bl[(Maxm<<1)+Maxn];
inline int query(int id,int l,int r){
    int ct=0,tt=0;
    if(r-l+1>=2*S_val){
        int L=bl[l]+(bl[l-1]==bl[l]);
        int R=bl[r]-(bl[r+1]==bl[r]);
        for(int i=l;i<bg[L];i++)ct+=(cnt[i]>0),tt+=cnt[i];
        for(int i=ed[R]+1;i<=r;++i)ct+=(cnt[i]>0),tt+=cnt[i];
        for(int i=L;i<=R;i++)ct+=bl_ans[i],tt+=bl_cnt[i];
    }else{
        for(int i=l;i<=r;i++)ct+=(cnt[i]>0),tt+=cnt[i];
    }
    ans1[id]=tt;ans2[id]=ct;
}
int main(){
    n=read(),m=read();S_qry=sqrt(n)+1;
    for(int i=1;i<=n;i++)a[i]=read(),lsh[++tot]=a[i];
    for(int i=1;i<=m;i++){
        qry[i].l=read();qry[i].r=read();qry[i].bl=(qry[i].l-1)/S_qry+1;
        lsh[++tot]=(qry[i].a=read());lsh[++tot]=(qry[i].b=read());
        qry[i].id=i;
    }
    sort(qry+1,qry+m+1);sort(lsh+1,lsh+tot+1);
    tot=unique(lsh+1,lsh+tot+1)-lsh-1;S_val=sqrt(tot)+1;
    for(int i=1;i<=n;i++)a[i]=lower_bound(lsh+1,lsh+tot+1,a[i])-lsh-1;
    for(int i=1;i<=m;i++){
        qry[i].a=lower_bound(lsh+1,lsh+tot+1,qry[i].a)-lsh-1;
        qry[i].b=lower_bound(lsh+1,lsh+tot+1,qry[i].b)-lsh-1;
    }
    for(int i=1;i<=tot;i++)bl[i]=(i-1)/S_val+1;
    for(int i=1,h=1;i<=tot;i+=S_val,++h){
        bg[h]=i;ed[h]=min(tot,i+S_val-1);
    }
    int L=1,R=0;
    for(int i=1;i<=m;i++){
        int l=qry[i].l,r=qry[i].r;
        while(L>l){
            --L;
            if(!cnt[a[L]]++)++bl_ans[bl[a[L]]];
            ++bl_cnt[bl[a[L]]];
        }
        while(L<l){
            if(!--cnt[a[L]])--bl_ans[bl[a[L]]];
            --bl_cnt[bl[a[L]]];
            ++L;
        }
        while(R<r){
            ++R;
            if(!cnt[a[R]]++)++bl_ans[bl[a[R]]];
            ++bl_cnt[bl[a[R]]];
        }
        while(R>r){
            if(!--cnt[a[R]])--bl_ans[bl[a[R]]];
            --bl_cnt[bl[a[R]]];
            --R;    
        }
        query(qry[i].id,qry[i].a,qry[i].b);
    }
    for(int i=1;i<=m;i++)W(ans1[i]),putchar(' '),W(ans2[i]),putchar('\n');
}