CF 475D.CGCDSSQ ST表 + 二分

题目链接

题意：

给定长度为$n$的正整数序列$a$。
有$m$组询问，每组询问包括一个正整数$x$。
对每个询问计算共有多少个区间$[l,r]$，满足$a_l,a_{l+1},\dots ,a_r$的最大公约数为$x$，即$gcd(a_l,a_{l+1},\dots ,a_r)=x$。

数据范围：

题解：

1.首先，我们使用$ST$表存储区间之间的最大公约数。

2.对每个询问$x$，枚举左端点$L$。

①对每个左端点$L$，在最大公约数为$g$的情况下，计算出满足$gcd([L,R])=g$的最大的$R$，$g$初始化为$st[L][0]$。

②那么对于最大公约数为$g$，且左端点为$L$的情况下，满足条件的区间个数为$R-L+1$。接着我们将$g$更新为$gcd([L,R+1])$，重复之前的操作，直到$R>n$。

由于区间的最大公约数随着区间右端点的不断增加，具有不增加的性质，因此，我们可以对右端点$R$进行二分查找。

此时，$gcd(a,b)=min(a,b)且gcd(a,b)\le \frac{min(a,b)}{2}$，此时，$gcd$最多只会变化$logn$次，也就是第$2$步最多只会执行$logn$次。

我们在$O(n)$枚举左端点的同时，二分查找满足条件的最大右端点，因此时间复杂度为$O(nlogn)$。

实现细节见代码：

#include <bits/stdc++.h>
using namespace std;
#define endl '\n'
#define int long long
typedef long long ll;
const int inf = 0x3f3f3f3f;
const int MAXN = 1e5 + 10;
const int mod = 1e9 + 7;
int st[MAXN][30], logn[MAXN], a[MAXN];
map<int, int> ans;
void init() {
    logn[1] = 0;
    for (int i = 2; i < MAXN; i++) {
        logn[i] = logn[i / 2] + 1;
    }
}
void get_st(int n) {
    for (int j = 1; j <= 25; j++) {
        for (int i = 1; i + (1 << j) - 1 <= n; i++) {
            st[i][j] = __gcd(st[i][j - 1], st[i + (1 << (j - 1))][j - 1]);
        }
    }
}
int query(int l, int r) {
    int k = logn[r - l + 1];
    return __gcd(st[l][k], st[r - (1 << k) + 1][k]);
}
int check(int g, int L, int l, int n) {
    int r = n;
    while (l < r) {
        int mid = l + r >> 1;
        if (query(L, mid) != g) {
            r = mid;
        }
        else {
            l = mid + 1;
        }
    }
    if (query(L, l) != g) l--; // 注意特判当前l是否符合要求
    return l;
}
void solve(int l, int n) {
    int r = l, g = st[l][0];
    while (r <= n) {
        int pre = r;
        r = check(g, l, r, n); // 二分满足区间公约数为g的最大右端点
        ans[g] += r - pre + 1;
        g = query(l, r + 1);
        r++;
    }
}
signed main() 
{
    ios::sync_with_stdio(false);
    cin.tie(0), cout.tie(0);
    int n;
    cin >> n;
    for (int i = 1; i <= n; i++) {
        cin >> a[i];
        st[i][0] = a[i];
    }
    init();
    get_st(n);
    for (int i = 1; i <= n; i++) { // 枚举左端点
        solve(i, n);
    }
    int q;
    cin >> q;
    while (q--) {
        int x;
        cin >> x;
        cout << ans[x] << endl;
    }
    return 0;
}