結果

問題 No.1036 Make One With GCD 2
コンテスト
ユーザー Shibuyap
提出日時 2020-05-29 16:47:10
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 2,917 bytes
コンパイル時間 2,764 ms
コンパイル使用メモリ 193,428 KB
最終ジャッジ日時 2025-01-10 16:08:50
ジャッジサーバーID
（参考情報） 		judge5 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 19 TLE * 22
権限があれば一括ダウンロードができます

ソースコード

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 << 19;

// セグメント木を持つグローバル配列
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);
    }
}


int main(){
    ll n;
    cin >> n;
    init(n);
    ll a[n];
    rep(i,n)cin >> a[i];

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

    ll ans = 0;

    rep(i,n){
        int l = i, r = n;
        if(query(i,r,0,0,nn) != 1)continue;
        if(a[i] == 1){
            ans += n - i;
            continue;
        }
        while(l + 1 < r){
            int m = (l + r) / 2;
            if(query(i,m,0,0,nn) == 1){
                r = m;
            }else{
                l = m;
            }
        }
        ans += n + 1 - r;
    }

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