結果
問題 | No.2075 GCD Subsequence |
ユーザー |
![]() |
提出日時 | 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)