結果

問題 No.1036 Make One With GCD 2
ユーザー srjywrdnprkt
提出日時 2024-02-29 19:33:27
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 391 ms / 2,000 ms
コード長 1,739 bytes
コンパイル時間 2,756 ms
コンパイル使用メモリ 203,000 KB
最終ジャッジ日時 2025-02-19 22:01:56
ジャッジサーバーID
(参考情報)
judge4 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>

using namespace std;
using ll = long long;

template<class S, S (*op)(S, S), S (*e)()>struct SegTree {
    private:
        vector<S> seg; int N = 1;
        S _prod (int l, int r, int idx, int bitl, int bitr) const{
            if (r < bitl || l > bitr) return e();
            if (l <= bitl && bitr <= r) return seg[idx];

            ll bitm = (bitl+bitr)/2;
            return op(_prod(l, r, idx*2+1, bitl, bitm), _prod(l, r, idx*2+2, bitm+1, bitr));
        }

    public:
        SegTree (ll n) : SegTree(vector<S>(n, e())) {}
        SegTree (const vector<S> &v){
            ll n = v.size(); while (N < n) N *= 2; seg.resize(N*2-1, e());
            for (ll i=0; i<n; i++) seg[i+N-1] = v[i];
            for (ll i=N-2; i>=0; i--) seg[i] = op(seg[i*2+1], seg[i*2+2]);
        }
        void set(int loc, S val){
            loc += N-1; seg[loc] = val;
            while (loc != 0) loc = (loc-1)/2, seg[loc] = op(seg[loc*2+1], seg[loc*2+2]);
        }
        S prod (int l, int r) const {return _prod(l, r, 0, 0, N-1);}
        S all_prod() const {return seg[0];}
        S get (int i) const{return seg[i+N-1];}
        void show() const{
            for (int i=N-1; i<N*2-1; i++) cout << seg[i] << " ";
            cout << endl;
        }
};

//Range GCD Query
using S = ll;
S op(S a, S b){return gcd(a, b);}
S e(){return 0;}
 
int main(){
    cin.tie(nullptr);
    ios_base::sync_with_stdio(false);
 
    ll N, ans=0, r=0;
    cin >> N;
    vector<S> v(N);
    for (int i=0; i<N; i++) cin >> v[i];
    SegTree<S, op, e> tree(v);

    for (int l=0; l<N; l++){
        while(tree.prod(l, r) != 1 && r < N) r++;
        ans += N-r;
        if (l == r) r++;
    }

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