結果
| 問題 | 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 |
ソースコード
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)
rlangevin