同余
题目大意
给你一个数组,多次询问,给你给你 l,r,p,q,问你数组 l~r 的位置中有多少个数满足对 p 取模得到的是 q。
思路
考虑从部分分入手,发现如果数组的数小于 100,那我们可以用一个前缀和来搞。
具体一下就是把操作离线,然后分成
1
∼
r
1\sim r
1∼r 的答案减去
1
∼
l
−
1
1\sim l-1
1∼l−1 的答案,把它们按右端点排序。
然后你每到一个新的右端点,你就把这个新的数组中的数丢进去处理。
至于怎么处理,你有一个
n
u
m
i
,
j
num_{i,j}
numi,j 为模
i
i
i 为
j
j
j 的数有多少个,你就枚举模数,把对应的位置加一加。
然后结果就直接拿
n
u
m
num
num 数组里的。
接着考虑
10000
10000
10000 的怎么搞。暴力枚举可以的数然后看有多少个显然不行,但我们考虑暴力为什么不行。
那你发现,如果模数很大,那可能的数就会很少,那这个时候暴力是可以过的。
然后如果模数很小,那可能的数就会很多,那这个时候暴力就炸了。
然后你考虑模数小的时候用我们的前缀和,模数大的时候暴力。
然后你看到这个个数的关系时除数,不难想到这个分界点就是
a
i
\sqrt{a_i}
ai,正好就是我们前面暴力刚好可以过的
100
100
100。
然后就可以了。
代码
#include<cstdio>
#include<algorithm>
using namespace std;
struct node {
int p, q, x;
int num, op;
}q[200001];
int n, m, a[100001];
int l, r, p, Q, ans[100001];
int num[101][101], number[10001];
bool cmp(node x, node y) {
return x.x < y.x;
}
int main() {
// printf("%.3lf", sizeof(t) / 1024.0);
scanf("%d %d", &n, &m);
for (int i = 1; i <= n; i++) {
scanf("%d", &a[i]);
}
for (int i = 1; i <= m; i++) {
scanf("%d %d %d %d", &l, &r, &p, &Q);
q[i * 2 - 1] = (node){p, Q, r, i, 1};
q[i * 2] = (node){p, Q, l - 1, i, -1};
}
sort(q + 1, q + m * 2 + 1, cmp);
for (int i = 1; i <= 2 * m; i++) {
if (q[i].x != q[i - 1].x) {//得出预处理
for (int j = q[i - 1].x + 1; j <= q[i].x; j++) {
for (int k = 1; k <= 100; k++)
num[k][a[j] % k]++;
number[a[j]]++;
}
}
if (q[i].p <= 100) {//小于 100 直接用预处理的
ans[q[i].num] += q[i].op * num[q[i].p][q[i].q];
}
else {//大于 100 直接暴力搞
for (int j = q[i].q; j <= 10000; j += q[i].p)
ans[q[i].num] += q[i].op * number[j];
}
}
for (int i = 1; i <= m; i++)
printf("%d\n", ans[i]);
return 0;
}