結果

問題 No.1036 Make One With GCD 2
ユーザー hitonanode
提出日時 2020-04-25 21:55:14
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 318 ms / 2,000 ms
コード長 1,684 bytes
コンパイル時間 2,307 ms
コンパイル使用メモリ 202,656 KB
最終ジャッジ日時 2025-01-10 01:41:55
ジャッジサーバーID
（参考情報） 		judge3 / judge4
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 41
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;
using lint = long long int;
struct fast_ios { fast_ios(){ cin.tie(0); ios::sync_with_stdio(false); cout << fixed << setprecision(20); }; } fast_ios_;

// Static sequence sparse table
// Complexity: O(NlogN) for precalculation, O(1) per query
template<typename T, typename F>
struct SparseTable {
    int N, lgN;
    T defaultT;
    F func;
    std::vector<std::vector<T>> data;
    std::vector<int> lgx_table;
    SparseTable(F func) : func(func) {}
    SparseTable(const std::vector<T> &sequence, T defaultT, F func) : N(sequence.size()), defaultT(defaultT), func(func)
    {
        lgx_table.resize(N + 1);
        for (int i = 2; i < N + 1; i++) lgx_table[i] = lgx_table[i >> 1] + 1;
        lgN = lgx_table[N] + 1;
        data.assign(lgN, std::vector<T>(N, defaultT));
        data[0] = sequence;
        for (int d = 1; d < lgN; d++) {
            for (int i = 0; i + (1 << d) <= N; i++) {
                data[d][i] = func(data[d - 1][i], data[d - 1][i + (1 << (d - 1))]);
            }
        }
    }
    T get(int l, int r) { // [l, r), 0-indexed
        assert(l >= 0 and r <= N);
        if (l >= r) return defaultT;
        int d = lgx_table[r - l];
        return func(data[d][l], data[d][r - (1 << d)]);
    }
};

int main()
{
    int N;
    cin >> N;
    vector<lint> A(N);
    for (auto &a : A) cin >> a;
    auto f = [](lint l, lint r) -> lint { return gcd(l, r); };
    SparseTable<lint, decltype(f)> st(A, 0, f);
    lint ret = 0;
    int r = 1;
    for (int l = 0; l < N; l++)
    {
        if (l == r) r++;
        while (r <= N and st.get(l, r) > 1) r++;
        ret += N - r + 1;
    }
    cout << ret << '\n';
}
0