JZOJ4744. 【NOIP2016提高A组模拟9.2】同余 一类比较好的分类思想+根号算法

题目大意

给你一个长度为$n$的序列，有$m$组询问，每组询问给你$l,r,p,q$表示要求你输出有多少个$i$满足$l\leq i \leq r$且$a_i\equiv q （\bmod p）$。

$n \leq 10^4$
$m \leq 10^5$
$0 \leq q < p \leq 10^4$
$a_i \leq 10^4$

解题思路

首先对于一个区间$[l,r]$的询问，我们显然可以用$[1,r]$中符合要求的个数减去$[1,l-1]$中符合要求的个数，这样就可以将一个询问拆成两个询问，然后从$1$到$n$扫一遍统计答案即可。

对于一个模数$p$，我们很难直接处理询问，那么我们考虑将$p$分为两种情况讨论。
1.$p \leq \sqrt{max_{a_i}}$：对于这种情况由于$p$最多只有$100$个，那么我们可以直接用数组$g[i]][j]$表示模$i$等于$j$的数有多少个$j\leq \sqrt{max_{a_i}}, i \leq j$，而扫的时候只会扫$n$个位置，维护这个数组的时候没加入一个新的$a_k$直接更新每个$i$对应的位置$j$，询问是直接$O(1)$。这部分的复杂度就是$O(n* \sqrt{max_{a_i}})$。

2.$p \geq \sqrt{max_{a_i}}$：其实每次询问相当于找都多少个$a_k$满足$a_k=p*t+q$，由于$p \geq \sqrt{max_{a_i}}$，所以符合要求的$t$最多只有$\sqrt{max_{a_i}}$个那么我们可以开一个数组$num_i$记录数组中有多少个数等于$i$，每次修改直接$O(1)$更改，每次询问则枚举$t$，累计$num_t$的个数。这部分的复杂度$O(n* \sqrt{max_{a_i}})$。

结合两种思想总的复杂度就是$O(n* \sqrt{max_{a_i}})$。

程序

#include <cstring>
#include <cstdio>
#include <algorithm>

using namespace std;

const int MAXN = 1e5 + 5, MAXM = 1e4 + 5;
const int Lim = 100;

struct Node {
    int side, ord, p, q, val;
    Node (int a, int b, int c, int d, int e) {side = a, ord = b, p = c, q = d, val = e;}
    Node () {}
};

Node Q[MAXN * 2];
int n, m, Max, num[MAXM], a[MAXN], ans[MAXN], g[Lim][Lim];

bool Cmp(Node A, Node B) {
    return A.side < B.side;
}

int main() {
    scanf("%d%d", &n, &m);
    for (int i = 1; i <= n; i ++) { 
        scanf("%d", &a[i]);
        Max = max(Max, a[i]);
    }
    int tot = 0;
    for (int i = 1; i <= m; i ++) {
        int l, r, p, q;
        scanf("%d%d%d%d", &l, &r, &p, &q);
        Q[++ tot] = Node(l - 1, i, p, q, -1);
        Q[++ tot] = Node(r, i, p, q, 1);
    }
    sort(Q + 1, Q + tot + 1, Cmp);
    int i = 1, j = 1;
    for (; Q[j].side < 1; j ++);
    for (; j <= tot; i ++) { 
        num[a[i]] ++;
        for (int k = 1; k < Lim; k ++) g[k][a[i] % k] ++;
        while (j <= tot && Q[j].side == i) {
            if (Q[j].p < Lim) {
                ans[Q[j].ord] += Q[j].val * g[Q[j].p][Q[j].q];
            } else {
                int t = Q[j].q;
                for (; t <= Max; t += Q[j].p) 
                    ans[Q[j].ord] += Q[j].val * num[t];
            }
            j ++;
        }
    }
    for (int i = 1; i <= m; i ++) 
        printf("%d\n", ans[i]);
}