結果

問題 No.1036 Make One With GCD 2
ユーザー 里旬
提出日時 2020-04-24 21:54:12
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,181 ms / 2,000 ms
コード長 2,215 bytes
コンパイル時間 2,226 ms
コンパイル使用メモリ 194,852 KB
最終ジャッジ日時 2025-01-09 23:38:35
ジャッジサーバーID
(参考情報)
judge5 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff #

#include <bits/stdc++.h>
using namespace std;

template <typename T, size_t N, T ident=T{}>
class disjoint_sparse_table{
    bool c;
    static constexpr size_t stage = N < 2 ? 1 : std::numeric_limits<unsigned long long>::digits - __builtin_clzll(N-1);
    T *arr[stage];
    T (*fn)(T, T);
    void construct(){
        size_t w = 2, l, r;
        for(size_t s=2;s<=stage;++s){
            w <<= 1;
            for(size_t i=0;i<N;i+=w){
                l = i; r = std::min(N - 1, i + w / 2 - 1);
                arr[stage-s][r] = arr[stage-1][r];
                for(size_t j=r;j>l;--j) arr[stage-s][j-1] = fn(arr[stage-1][j-1], arr[stage-s][j]);
                l = i + w / 2; r = std::min(N - 1, i + w -1);
                if(l >= N || l > r) continue;
                arr[stage-s][l] = arr[stage-1][l];
                for(size_t j=l+1;j<=r;++j) arr[stage-s][j] = fn(arr[stage-s][j-1], arr[stage-1][j]);
            }
        }
        c = true;
    }
public:
    disjoint_sparse_table(T(*f)(T, T)): c(false), fn(f){
        for(size_t s=0;s<stage;++s) arr[s] = new T[N];
    }
    ~disjoint_sparse_table(){ for(size_t s=0;s<stage;++s) delete[] arr[s]; }

    T& operator[](size_t i) & noexcept { c = false; return arr[stage-1][i]; }
    T get(size_t i) const noexcept { return arr[stage-1][i]; }

    T query(size_t left, size_t right) noexcept{
        if(left >= right || right > N) return ident;
        if(left+1 == right) return arr[stage-1][left];
        if(!c) construct();
        --right;
        auto s = std::numeric_limits<unsigned long long>::digits - __builtin_clzll(left ^ right);
        return fn(arr[stage-s][left], arr[stage-s][right]);
    }
};

int main(){
    cin.tie(nullptr);
    ios::sync_with_stdio(false);

    int n;
    disjoint_sparse_table<int64_t, 500000, 0> dst{[](auto x, auto y){ return gcd(x, y); }};
    cin >> n;
    for(int i=0;i<n;++i) cin >> dst[i];
    for(int i=n;i<500000;++i) dst[i] = 0;
    int64_t ans = 0;

    for(int i=0;i<n;++i){
        int l=i-1, r=n;
        while(r-l > 1){
            int m = (l+r)/2;
            if(dst.query(i, m+1) == 1) r = m;
            else l = m;
        }
        ans += n-r;
    }
    cout << ans << endl;
    
    return 0;
}
0