結果

問題 No.2075 GCD Subsequence
ユーザー rlangevin
提出日時 2024-04-13 16:51:14
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 2,059 ms / 4,000 ms
コード長 932 bytes
コンパイル時間 530 ms
コンパイル使用メモリ 82,560 KB
実行使用メモリ 201,484 KB
最終ジャッジ日時 2024-10-03 00:15:40
合計ジャッジ時間 31,111 ms
ジャッジサーバーID
(参考情報)
judge3 / judge1
このコードへのチャレンジ
(要ログイン)
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff #

from math import ceil, sqrt
def moebius_table(n):
    p = list(range(n + 1))
    for x in range(2, ceil(sqrt(n)) + 1):
        if p[x] == x:
            for y in range(x*x, n + 1, x):
                p[y] = x
            for y in range(x*x, n + 1, x*x):
                p[y] = 0
    res = [0] * (n + 1)
    res[1] = 1
    for x in range(2, n + 1):
        if p[x]:
            res[x] = -res[x//p[x]]

    return res


N = int(input())
A = list(map(int, input().split()))
S = set(A)
M = max(A) + 5
dp = [0] * M
g = [0] * M
s = 0
mod = 998244353
d = [[] for i in range(M)]
for i in range(1, M):
    for j in range(i, M, i):
        if j in S:
            d[j].append(i)

m = moebius_table(M)

for a in A:
    temp = 0
    for dd in d[a]:
        temp += m[dd] * g[dd]
    v = 1 + s - temp
    s += v
    dp[a] += v
    s %= mod
    dp[a] %= mod
    for dd in d[a]:
        g[dd] += v
        g[dd] %= mod
        
print(sum(dp) % mod)
0