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;}