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

原题链接
题意:
给一个正数数组,每一步操作会把所有ai变成gcd(ai, ai+1),对于an=gcd(an, a1),求最少多少步之后会使得所有数都相同。

思路:
答案具有单调性,所以可以二分答案,查询操作可以用线段树或稀疏表维护一下区间最大公因数,题目给了四秒,基本怎么写都不会超时,我线段树跑了1169ms,稀疏表跑了549ms,试了一下其他人的稀疏表可以跑到300多ms。

线段树

#include<bits/stdc++.h>
#define LL long long
#define INF INT64_MAX
#define MOD 998244353
#define ls rt << 1
#define rs rt << 1 | 1
using namespace std;
typedef pair<int,int>pa;
const int N = 2e5+7;
int a[N<<1], SegTree[N<<3];
void build(int rt, int l, int r)
{
    if(l==r)
    {
        SegTree[rt] = a[l];
        return;
    }
    int mid = l+r>>1;
    build(ls, l, mid);
    build(rs, mid+1, r);
    SegTree[rt] = __gcd(SegTree[ls], SegTree[rs]);
}
int query(int rt, int l, int r, int ql, int qr)
{
    if(l>=ql && r<=qr)  return SegTree[rt];
    int mid = l+r>>1;
    int res = 0;
    if(mid>=ql) res = __gcd(res, query(ls, l, mid, ql, qr));
    if(mid< qr) res = __gcd(res, query(rs, mid+1, r, ql, qr));
    return res;
}
int main()
{
    int t, n;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        int f = 0;
        for(int i = 1;i <= n;i++)
        {
            scanf("%d", &a[i]);
            f = __gcd(f, a[i]);
        }
        for(int i = 1;i <= n;i++)
        {
            a[i]/=f;
            a[i+n]=a[i];
        }
        n<<=1;
        build(1, 1, n);
        int ans = 0;
        for(int i = 1;i <= n/2;i++)
        {
            if(a[i]==1) continue;
            int l = i, r = n;
            while(l<r)
            {
                int mid = l+r>>1;
                if(query(1, 1, n, i, mid)==1) r = mid;
                else l = mid+1;
            }
            ans = max(ans, r-i);
        }
        printf("%d\n", ans);
    }
    return 0;
}

稀疏表

#include<bits/stdc++.h>
#define LL long long
#define INF INT64_MAX
#define MOD 998244353
#define ls rt << 1
#define rs rt << 1 | 1
using namespace std;
typedef pair<int,int>pa;
const int N = 2e5+7;
int a[N<<1], dp[N<<1][30], n;
void init()
{
    for(int j = 0;j < 18;j++)
    {
        for(int i = 1;i+(1<<j)-1 <= n;i++)
        {
            if(!j) dp[i][j] = a[i];
            else dp[i][j] = __gcd(dp[i][j-1], dp[i+(1<<j-1)][j-1]);
        }
    }
}
int query(int l, int r)
{
    int len = r-l+1;
    int k = log(len)/log(2);
    return __gcd(dp[l][k], dp[r-(1<<k)+1][k]);
}
int main()
{
    int t;
    scanf("%d", &t);
    while(t--)
    {
        scanf("%d", &n);
        int f = 0;
        for(int i = 1;i <= n;i++)
        {
            scanf("%d", &a[i]);
            f = __gcd(f, a[i]);
        }
        for(int i = 1;i <= n;i++)
        {
            a[i]/=f;
            a[i+n]=a[i];
        }
        n<<=1;
        init();
        int ans = 0;
        for(int i = 1;i <= n/2;i++)
        {
            if(a[i]==1) continue;
            int l = i, r = n;
            while(l<r)
            {
                int mid = l+r>>1;
                if(query(i, mid)==1) r = mid;
                else l = mid+1;
            }
            ans = max(ans, r-i);
        }
        printf("%d\n", ans);
    }
    return 0;
}

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值