結果

問題 No.1036 Make One With GCD 2
ユーザー ShibuyapShibuyap
提出日時 2020-05-29 17:55:38
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,692 ms / 2,000 ms
コード長 2,812 bytes
コンパイル時間 3,353 ms
コンパイル使用メモリ 193,172 KB
最終ジャッジ日時 2025-01-10 16:15:07
ジャッジサーバーID
(参考情報)
judge2 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
#define rep(i,n) for(int i = 0; i < (n); ++i)
#define rrep(i,n) for(int i = 1; i <= (n); ++i)
#define drep(i,n) for(int i = (n)-1; i >= 0; --i)
#define srep(i,s,t) for (int i = s; i < t; ++i)
using namespace std;
typedef pair<int,int> P;
typedef long long int ll;
#define dame { puts("-1"); return 0;}
#define yn {puts("Yes");}else{puts("No");}

const int MAX_N = 1 << 20;

// セグメント木を持つグローバル配列
ll nn, dat[2 * MAX_N - 1];

// 関数
long long int GCD(long long int x, long long int y){
    x = abs(x);
    y = abs(y);
    if(x == 0) return abs(y);
    if(y == 0) return abs(x);
    if(x>y){
        long long int swap = x;
        x = y;
        y = swap;
    }
    while(true){
        if(y%x==0){
            return abs(x);
        }else{
            long long int amari = y%x;
            y = x;
            x = amari;
        }
    }
}

// function
ll func(ll x, ll y){
    return GCD(x, y);
}

// 単位元を入れる
ll init_val(){
    return 0;
}

// 初期化
void init(int n_){
    // n_が小さいときの変な挙動を防ぐ
    if(n_ < 10)n_ = 10;
    // 簡単のため要素数を2のべき乗に
    nn = 1;
    while(nn < n_)nn *= 2;

    // 全ての値を単位元に
    for(int i = 0; i < 2 * nn - 1; i++)dat[i] = init_val(); // 適宜変える
}

// k番目の値(0-indexed)をaに変更
void update(int k, ll a){
    // 葉の接点
    k += nn - 1;
    dat[k] = a;
    // 登りながら更新
    while(k > 0){
        k = (k - 1) / 2;
        dat[k] = func(dat[k * 2 + 1], dat[k * 2 + 2]);
    }
}

// [a,b)の最小値を求める
// 後ろのほうの引数は、計算の簡単のための引数。
// kは節点の番号、l, rはその節点が[l, r)に対応づいていることを表す。
// したがって、外からはquery(a, b, 0, 0, nn)として呼ぶ。
ll query(int a, int b, int k, int l, int r){
    // [a, b)と[l, r)が交差しなければ、単位元
    if (r <= a || b <= l)return init_val();

    // [a, b)が[l, r)を完全に含んでいれば、この節点の値
    if (a <= l && r <= b)return dat[k];
    else {
        // そうでなければ、2つの子の最小値
        ll vl = query(a, b, k * 2 + 1, l, (l + r) / 2);
        ll vr = query(a, b, k * 2 + 2, (l + r) / 2, r);
        return func(vl, vr);
    }
}

ll a[600000];

int main(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    ll n;
    cin >> n;
    init(n);
    rep(i,n)cin >> a[i];

    rep(i,n){
        update(i,a[i]);
    }

    ll ans = 0;
    int now = 0;

    rep(i,n){
        now = max(now, i);
        while(now < n && query(i,now+1,0,0,nn) != 1){
            now++;
        }

        if(now == n){
            break;
        }

        ans += n - now;

    }

    cout << ans << endl;
    return 0;
}

0