BZOJ3236 作业 [莫队算法][树状数组]

先用莫队离线对询问排序后维护两个树状数组，记录两个询问的答案，对于第二个询问，可以再开一个数组，记录每个数出现的次数。

#include<bits/stdc++.h>
#define lowbit(x) (x&-x)
using namespace std;
const int N = 100005;
int n,m,a[N],ans1[N*10],ans2[N*10],cf[N],cs[N],num[N],pos[N],sqrtn;
struct data{
    int l,r,a,b,id;
    inline bool operator < (const data rhs)const{
        return pos[l]<pos[rhs.l]||(pos[l]==pos[rhs.l]&&r<rhs.r);
    }
}node[N*10];
inline void updatef(int rt,int val){while(rt<=n)cf[rt]+=val,rt+=lowbit(rt);}
inline void updates(int rt,int val){while(rt<=n)cs[rt]+=val,rt+=lowbit(rt);}
inline int queryf(int rt){int ans=0;while(rt)ans+=cf[rt],rt-=lowbit(rt);return ans;}
inline int querys(int rt){int ans=0;while(rt)ans+=cs[rt],rt-=lowbit(rt);return ans;}
void read(int &res){  
    res=0;char c=getchar();for(;!isdigit(c);c=getchar());  
    for(;isdigit(c);res=(res<<1)+(res<<3)+c-48,c=getchar());  
}
int main(){
    read(n),read(m),sqrtn=sqrt(n);
    for(register int i=1;i<=n;++i)read(a[i]),pos[i]=i/sqrtn;
    for(register int i=1;i<=m;i++){
        read(node[i].l),read(node[i].r),
        read(node[i].a),read(node[i].b);
        node[i].id=i;
    }
    sort(node+1,node+1+m);
    int L=1,R=1;
    updatef(a[1],1),updates(a[1],1);
    ++num[a[1]];
    for(register int i=1;i<=m;i++){
        while(L>node[i].l){
            L--;
            updatef(a[L],1);
            if(!num[a[L]])updates(a[L],1);
            num[a[L]]++;
        }
        while(R<node[i].r){
            R++;
            updatef(a[R],1);
            if(!num[a[R]])updates(a[R],1);
            num[a[R]]++;
        }
        while(L<node[i].l){
            updatef(a[L],-1);
            if(!--num[a[L]])updates(a[L],-1);
            L++;
        }
        while(R>node[i].r){
            updatef(a[R],-1);
            if(!--num[a[R]])updates(a[R],-1);
            R--;
        }
        ans1[node[i].id]=queryf(node[i].b)-queryf(node[i].a-1);
        ans2[node[i].id]=querys(node[i].b)-querys(node[i].a-1);
    }
    for(register int i=1;i<=m;i++)
        printf("%d %d\n",ans1[i],ans2[i]);
    return 0;
}