結果

問題 No.1036 Make One With GCD 2
ユーザー ShibuyapShibuyap
提出日時 2020-05-29 16:49:35
言語 C++17
(gcc 12.3.0 + boost 1.83.0)
結果
TLE  
実行時間 -
コード長 2,971 bytes
コンパイル時間 1,930 ms
コンパイル使用メモリ 196,012 KB
実行使用メモリ 29,080 KB
最終ジャッジ日時 2024-04-23 13:57:36
合計ジャッジ時間 10,064 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge1
このコードへのチャレンジ
（要ログイン）

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
testcase_22 -- -
testcase_23 -- -
testcase_24 -- -
testcase_25 -- -
testcase_26 -- -
testcase_27 -- -
testcase_28 -- -
testcase_29 -- -
testcase_30 -- -
testcase_31 -- -
testcase_32 -- -
testcase_33 -- -
testcase_34 -- -
testcase_35 -- -
testcase_36 -- -
testcase_37 -- -
testcase_38 -- -
testcase_39 -- -
testcase_40 -- -
testcase_41 -- -
testcase_42 -- -
testcase_43 -- -
testcase_44 -- -
権限があれば一括ダウンロードができます

ソースコード

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(){
    ios_base::sync_with_stdio(false);
    cin.tie(0);
    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