[hackerrank]Range Modular Queries

题目大意

区间询问模x等于y的值的个数
值域是[0,40000]
允许离线

平衡结合

对于x<=200的询问，因为x很小所以y也很小，这时候可以考虑使用分块。
将序列分块，然后处理sum[i,j,k]表示前i块中模j等于k的值有多少个。
询问时暴力扫剩余部分。
对于x>200的询问，因为x较大所以符合条件的数很小，这时候可以考虑使用莫队。
维护一个桶，每次询问只要看一下对应符合的数字有多少个即可。
不得不说，这是一道根号算法的好题，将三种根号算法结合到了一道题上。

#include<cstdio>
#include<algorithm>
#include<cmath>
#define fo(i,a,b) for(i=a;i<=b;i++)
using namespace std;
const int maxn=40000+10,mx=40000,B=200;
struct dong{
    int l,r,x,y,id;
    /*dong(){}
    dong (int _l,int _r,int _x,int _y,int _id){ l=_l;r=_r;x=_x;y=_y;id=_id; }*/
} ask[maxn];
int sum[B+10][B+10][B+10],belong[maxn],a[maxn],ans[maxn],b[maxn],sta[80];
int i,j,k,l,r,x,y,t,n,m,tot,top;
int read(){
    int x=0,f=1;
    char ch=getchar();
    while (ch<'0'||ch>'9'){
        if (ch=='-') f=-1;
        ch=getchar();
    }
    while (ch>='0'&&ch<='9'){
        x=x*10+ch-'0';
        ch=getchar();
    }
    return x*f;
}
int solve(int j,int k,int x,int y){
    int i,t=0,l=belong[j],r=belong[k];
    if (l==r){
        fo(i,j,k)
            if (a[i]%x==y) t++;
    }
    else{
        t=sum[r-1][x][y]-sum[l][x][y];
        fo(i,j,l*B)
            if (a[i]%x==y) t++;
        fo(i,(r-1)*B+1,k)
            if (a[i]%x==y) t++;
    }
    return t;
}
void ins(int x){
    b[a[x]]++;
}
void del(int x){
    b[a[x]]--;
}
int work(int x,int y){
    int k=y,t=0;
    while (k<=mx){
        t+=b[k];
        k+=x;
    }
    return t;
}
int main(){
    n=read();m=read();
    fo(i,1,n) a[i]=read();
    fo(i,1,n) belong[i]=(i-1)/B+1;
    fo(i,1,n)
        fo(j,1,B) sum[belong[i]][j][a[i]%j]++;
    fo(i,1,belong[n])
        fo(j,1,B)
            fo(k,0,j-1)
                sum[i][j][k]+=sum[i-1][j][k];
    fo(i,1,m){
        l=read();r=read();x=read();y=read();
        l++;r++;
        if (x<=B) ans[i]=solve(l,r,x,y);
        else ask[++top]=(dong){l,r,x,y,i};
    }
    l=1;r=0;
    fo(i,1,top){
        while (l<ask[i].l) del(l++);
        while (l>ask[i].l) ins(--l);
        while (r>ask[i].r) del(r--);
        while (r<ask[i].r) ins(++r);
        ans[ask[i].id]=work(ask[i].x,ask[i].y);
    }
    fo(i,1,m) printf("%d\n",ans[i]);
}