結果

問題 No.2075 GCD Subsequence
ユーザー rlangevinrlangevin
提出日時 2024-04-14 20:52:20
言語 PyPy3
(7.3.15)
結果
AC  
実行時間 1,261 ms / 4,000 ms
コード長 1,044 bytes
コンパイル時間 336 ms
コンパイル使用メモリ 82,304 KB
実行使用メモリ 203,996 KB
最終ジャッジ日時 2024-10-03 23:42:09
合計ジャッジ時間 20,564 ms
ジャッジサーバーID
（参考情報） 		judge1 / judge3
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 3
other AC * 28
権限があれば一括ダウンロードができます

ソースコード

diff # 

class Mobius():
    def __init__(self, N):
        self.L = list(range(N + 1))
        self.mobius = [1] * (N + 1)
        for i in range(2, N + 1):
            if i != self.L[i]:
                continue
            self.mobius[i] = -1
            for j in range(2 * i, N + 1, i):
                if self.L[j] == j:
                    self.L[j] = i
                if (j // i) % i == 0:
                    self.mobius[j] = 0
                else:
                    self.mobius[j] = - self.mobius[j]


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

m = Mobius(M).mobius

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