Codeforces Round #731 (Div. 3) F. Array Stabilization (GCD version) ST表+二分

题目链接

题目大意

给你n个数，问你多少次操作之后可以把这n个数变为相同的数。
定义一次操作如下，每个数与后一个数取gcd，如果是最后一个数那么就与第一个数取gcd来代替原来这个数。
例如
𝑎=[16,24,10,5]
operator
a=[gcd(16,24), gcd(24,10), gcd(10,5), gcd(5,16)] =[8,2,5,1].

解题思路

首先可以分析出，最后n个相同的数是最初的n个数取gcd，设为g
对于题目给的操作我们进行简单的分析可以发现，对于n个数任意一个数，他进行1次操作是与后面一个数去gcd，进行两次操作就是与后面两个数取gcd，进行x次操作就是与后面x个数取gcd，显然，一个数与后面n-1的数取gcd一定是g，所以答案的取值范围是[0,n-1]，我们二分答案即可。
如何判断mid次是否能满足要求呢，我们就对每个数进行mid次操作，即[i,i+mid]区间的数取gcd，如果任意一个取gcd不等于g，那说明mid次不满足要求。
我们发现gcd具有线性的性质，对于一个区间查询，我们就可以用线段树来维护，但是考虑到我们不需要修改，我们用ST表来维护更为简单快捷，最后的复杂度为O(nlogn)

代码

#include <bits/stdc++.h>
#define ll long long
#define qc ios::sync_with_stdio(false); cin.tie(0);cout.tie(0)
#define fi first
#define se second
#define PII pair<int, int>
#define PLL pair<ll, ll>
#define pb push_back
using namespace std;
const int MAXN = 4e5 + 7;
const int inf = 0x3f3f3f3f;
const ll INF = 0x3f3f3f3f3f3f3f3f;
const ll mod = 1e9 + 7;
int a[MAXN];
ll gcd(ll a, ll b){
    return b ? gcd(b, a%b) : a;
}
ll g;
int n;
ll st[MAXN][30];

void init(){
    for(int i = 1; i <= 2*n; i++)
        st[i][0] = a[i];
    for(int j = 1; (1 << j) <= 2*n; j++){
        for(int i = 1; i + (1 << j) <= 2*n; i++)
            st[i][j] = gcd(st[i][j-1], st[i + (1 << (j-1))][j-1]);
    }
}

ll search(int l, int r){
    int k = (int)(log(1.0*(r - l + 1)) / log(2.0));
    return gcd(st[l][k], st[r - (1 << k) + 1][k]);
}
int check(int x){
    for(int i = 1; i <= n; i++){
        if(g != search(i, i + x)) return 0;
    }
    return 1;
}

void solve(){
    cin >> n;
    for(int i = 1; i <= n; i++){
    	cin >> a[i];
        a[i+n] = a[i];
        if(i == 1) g = a[i];
        else g = gcd(g, a[i]);
    }
    init();
    int l = 0, r = n - 1;
    while(l < r){
        int mid = (l + r) >> 1;
        if(check(mid))
            r = mid;
        else
            l = mid + 1;
    }
    cout << l << endl;
}

int main()
{
	#ifdef ONLINE_JUDGE
    #else
       freopen("in.txt", "r", stdin);
       freopen("out.txt", "w", stdout);
    #endif

	qc;
    int T;
    cin >> T;
    // T = 1;
    while(T--){

        solve();
    }
	return 0;
}