洛谷P5629 【AFOI-19】区间与除法（思维，ST表）

P5629 【AFOI-19】区间与除法

原题链接

这题首先要明确一个结论，一个 $a[i]$ 如果对应了多个 $b[k]$ ，那么用最小的 $b[k]$ 一定最优。这个可以简单的证明一下，若最终答案里 $a[i]$ 与 $a[j]$ 对应的是同一个 $b[k]$ ，那么让他们同时继续除 $d$ ，除到最小的 $b[h]$ 并不影响答案。如果答案中 $a[i]$ 对应了唯一的 $b[k]$ ，那让其一直除 $d$ 至最小的 $b[h]$ 也不影响答案，也就是说，所有有 $b$ 对应的 $a[i]$ 都用最小的 $b$ 就是最优的。那么其实对于一个区间询问答案实际上就是在询问区间并集的集合元素个数，然后我们发现 $m\le 60$ ，我们就可以状压以后维护区间或来维护区间并集的元素个数。

#include <bits/stdc++.h>

using namespace std;

typedef long long ll;

const int N = 5e5 + 10;

int n, m, d, q, tot;

ll f[N][25];
int logn[N];

ll a[N], b[100], mn[N], t[N];

ll v[100], cnt;

void init() {
    logn[1] = 0;
    for (int i = 2; i <= n; ++i) {
        logn[i] = logn[i / 2] + 1;
    }
    for (int j = 1; j < 21; ++j) {
        for (int i = 1; i + (1 << j) - 1 <= n && i <= n; ++i) {
            f[i][j] = f[i][j - 1] | f[i + (1 << (j - 1))][j - 1];
        }
    }
}

int query(int l, int r) {
    int s = logn[r - l + 1];
    ll cur = f[l][s] | f[r - (1 << s) + 1][s];
    // cout << "======" << cur << endl;
    int ans = 0;
    while(cur) {
        if (cur & 1ll) ans++;
        cur >>= 1ll;
    }
    return ans;
}

int main() {
#ifndef ONLINE_JUDGE
    freopen("in.txt", "r", stdin);
    freopen("out.txt", "w", stdout);
#endif
    memset(t, -1, sizeof t);
    scanf("%d%d%d%d", &n, &m, &d, &q);
    for (int i = 1; i <= n; ++i) {
        scanf("%lld", &a[i]);
    }
    for (int i = 1; i <= m; ++i) {
        scanf("%lld", &b[i]);
    }
    sort(b + 1, b + 1 + m);
    for (int i = 1; i <= n; ++i) {
        int pos = -1;
        do {
            int idx = lower_bound(b + 1, b + 1 + m, a[i]) - b;
            if (idx >= 1 && idx <= m && b[idx] == a[i]) {
                pos = idx;
            } 
            a[i] /= d;
        }while(a[i]);
        if (~pos) f[i][0] = (1ll << pos);
        // cout << f[i][0] << ' ';
    }
    // cout << endl;
    init();
    while(q--) {
        int l, r;
        scanf("%d%d", &l, &r);
        printf("%d\n", query(l, r));
    }
}