ST表 + 二分 - CGCDSSQ - CodeForces - 475D

本文链接：https://blog.csdn.net/njuptACMcxk/article/details/106753537

ST表 + 二分 - CGCDSSQ - CodeForces - 475D

题意：

$给定长度为n的正整数序列a_1,a_2,...,a_n。$

$进行m组询问，每组询问包括一个正整数x_i，i∈[1,m]。$

$计算共有多少个区间[l,r]，满足a_l,a_{l+1},...,a_r的最大公约数为x，即gcd(a_l,a_{l+1},...,a_r)=x_i。$

Examples

Input

Output

Input

7
10 20 3 15 1000 60 16
10
1
2
3
4
5
6
10
20
60
1000

Output

数据范围：

$1 ≤ n ≤ 10^5，1 ≤ a_i ≤ 10^9，1 ≤ m ≤ 3 × 10^5，1 ≤ x_i ≤ 10^9\\Time\ limit:2000 ms，Memory\ limit:262144 kB$

分析：

$① 、首先用 S T 表存储区间 [l, r] 之间的正整数的最大公约数。$

$②、对每一次询问x_i，暴力枚举区间左端点L_i。$

$\qquad对每个左端点L_i，在最大公约数为t的情况下，计算满足gcd([L_i,R])=t的最大的R的值，t初始为a_i=f[L_i][0]。$

$\qquad则对最大公约数为t，且左端点为L的情况下，满足条件的区间数量为R-L+1。$

$\qquad接着将t更新为gcd([L,R+1])，重复操作，直至R>n。$

$补充：第 ② 步中，求最大 R 的过程可以二分。$

$因为，gcd(a,b)=min(a,b)或者gcd(a,b)≤\frac{min(a,b)}{2}，故gcd最多变化logn次，即第②步最多执行logn次。$

$这样，我们预处理所有左端点的情况，即执行第 ② 步 n 次，用 m a p 来统计每个最大公约数的满足条件的区间数量，$

$最后直接查询即可。时间复杂度 O (n l o g n) 。$

代码：

#include<iostream>
#include<cstdio>
#include<cstring>
#include<algorithm>
#include<map>

using namespace std;

const int N=1e5+10,M=20;

int n,m;
int pow_2[M],log_2[N];
int f[N][M];   
map<int,long long> cnt;

int gcd(int a,int b)
{
    return b ? gcd(b,a%b) : a;
}

void Init()
{
    for(int i=0;i<M;i++) pow_2[i]=1<<i;
    
    log_2[0]=-1;
    for(int i=1;i<N;i++) log_2[i]=log_2[i/2]+1;
}

void get_ST()
{
    for(int i=1;i<=log_2[n];i++)   
        for(int j=1;j+pow_2[i]-1<=n;j++)
                f[j][i]=gcd(f[j][i-1],f[j+pow_2[i-1]][i-1]);
}

int query(int l,int r)
{
    int k=log_2[r-l+1];
    return gcd(f[l][k],f[r-pow_2[k]+1][k]);
}

int bin(int x,int l,int L)   //二分以x为最大公约数，左端点在L处的区间的最大右端点R
{
    int r=n;
    while(l<r)
    {
        int mid=l+r+1>>1;
        if(query(L,mid)!=x) r=mid-1;
        else l=mid;
    }
    return l;
}

void pre_solve(int L)     //统计以L为左端点的所有区间，计算所有可能的最大公约数的满足条件的区间数量
{
    int t=f[L][0],R=L;
    while(R<=n)
    {
        int last=R;
        R=bin(t,R,L);
        cnt[t]+=R-last+1;
        R++;
        t=query(L,R);
    }
}

int main()
{    
    cin>>n;
    for(int i=1;i<=n;i++) scanf("%d",&f[i][0]);
    
    Init();
    get_ST();
    for(int i=1;i<=n;i++) pre_solve(i);
    
    cin>>m;
    int x;
    for(int i=1;i<=m;i++)
    {
        scanf("%d",&x);
        printf("%lld\n",cnt[x]);
    }
    
    return 0;
}