题目大意
区间询问模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]);
}