結果

問題 No.2075 GCD Subsequence
ユーザー kuronikuroni
提出日時 2022-09-16 22:18:24
言語 C++14
(gcc 13.3.0 + boost 1.87.0)
結果
AC  
実行時間 1,238 ms / 4,000 ms
コード長 1,164 bytes
コンパイル時間 1,678 ms
コンパイル使用メモリ 176,204 KB
実行使用メモリ 95,392 KB
最終ジャッジ日時 2024-12-21 21:32:51
合計ジャッジ時間 39,438 ms
ジャッジサーバーID
（参考情報） 		judge3 / judge2
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

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

using mint = modint998244353;

const int MX = 1E6 + 5;

int main() {
    ios_base::sync_with_stdio(false);
    cin.tie(nullptr);
    int n; cin >> n;
    vector<int> pr(MX);
    vector<int> pf(MX);
    vector<vector<int>> d(MX);
    for (int i = 2; i < MX; i++) {
        if (pr[i] == 0) {
            for (int j = i; j < MX; j += i) {
                pr[j] = i;
            }
            pf[i] = 1;
        } else {
            if ((i / pr[i]) % pr[i] != 0 && pf[i / pr[i]]) {
                pf[i] = -pf[i / pr[i]];
            } else {
                pf[i] = 0;
            }
        }
        if (pf[i]) {
            for (int j = i; j < MX; j += i) {
                d[j].push_back(i);
            }
        }
    }
    vector<mint> cnt(MX), dp(n);
    for (int i = 0; i < n; i++) {
        int u; cin >> u;
        mint ans = 1;
        for (int x : d[u]) {
            ans += cnt[x] * pf[x];
        }
        dp[i] = ans;
        for (int x : d[u]) {
            cnt[x] += ans;
        }
    }
    cout << accumulate(dp.begin(), dp.end(), mint(0)).val();
}
0