【JZOJ4744】【NOIP2016提高A组模拟9.2】同余

题目描述

输入

输出

样例输入

5 2

1 5 2 3 7

1 3 2 1

2 5 3 0

样例输出

2

1

数据范围

解法

题目允许离线，且没有修改操作。

考虑把一个询问拆分成两个形如”a b c”的询问，表示1~a这些数中mod b余c有多少个。

使用扫描线，把元素加入一个桶tong里面去。①

对于一个询问”a b c”，那么答案就是tong[c+kb] (k|c+kb<=10000)。

当b>=100时，询问答案最多耗时O(100)；

当b<100时，询问答案最多耗时O(10000)，考虑到b比较小，所以在①处额外维护一个二维桶xt[][]，xt[i][j]表示模数为i，余数为j的个数有多少个。

于是询问变成O(1)。

总的时间复杂度为O(n*maxa^0.5)。

代码

#include<iostream>
#include<cstdio>
#include<cstring>
#include<cmath>
#include<algorithm>
#define ll long long
#define sqr(x) ((x)*(x))
#define ln(x,y) int(log(x)/log(y))
using namespace std;
const char* fin="aP3.in";
const char* fout="aP3.out";
const int inf=0x7fffffff;
const int maxn=100007,maxa=10007,mina=107,maxv=maxn*2;
int n,m,i,j,k,l,o,tot,maxx,minx;
int a[maxn];
int ans[maxn][2];
int tong[maxa],xt[mina][mina];
struct visit{
int en,p,q,id,tp;
void init(int a,int b,int c,int d,int e){
en=a;p=b;q=c;id=d;tp=e;
}
void work(){
int i,j,k;
if (p>minx) for (j=q;j<=maxx;j+=p) ans[id][tp]+=tong[j];
else ans[id][tp]=xt[p][q];
}
}b[maxv];
bool cmp(visit i,visit j){
return i.en<j.en;
}
int main(){
scanf("%d%d",&n,&m);
for (i=1;i<=n;i++){
scanf("%d",&a[i]);
maxx=max(maxx,a[i]);
}
maxx=10000;
minx=int(sqrt(maxx));
for (i=1;i<=m;i++){
scanf("%d%d%d%d",&j,&k,&l,&o);
if (j>1) b[++tot].init(j-1,l,o,i,0);
b[++tot].init(k,l,o,i,1);
}
sort(b+1,b+tot+1,cmp);
k=1;
for (i=1;i<=n;i++){
tong[a[i]]++;
for (j=1;j<=minx;j++) xt[j][a[i]%j]++;
while (k<=tot && b[k].en==i) b[k++].work();
}
for (i=1;i<=m;i++){
j=ans[i][1]-ans[i][0];
printf("%d\n",j);
}
return 0;
}

启发

本题关键有两处，①是询问离线且没有修改，②是maxa<=10000；

①使得问题可以转化为简单的子问题；

②使得可以分类讨论来进行优化。

日后额外注意形如上述性质。