結果
問題 | No.2075 GCD Subsequence |
ユーザー |
![]() |
提出日時 | 2025-02-07 23:37:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 489 ms / 4,000 ms |
コード長 | 2,566 bytes |
コンパイル時間 | 1,357 ms |
コンパイル使用メモリ | 82,504 KB |
実行使用メモリ | 119,216 KB |
最終ジャッジ日時 | 2025-02-07 23:37:23 |
合計ジャッジ時間 | 11,512 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 28 |
ソースコード
import sys # sys.setrecursionlimit(200005) # sys.set_int_max_str_digits(200005) int1 = lambda x: int(x)-1 pDB = lambda *x: print(*x, end="\n", file=sys.stderr) p2D = lambda x: print(*x, sep="\n", end="\n\n", file=sys.stderr) def II(): return int(sys.stdin.readline()) def LI(): return list(map(int, sys.stdin.readline().split())) def LLI(rows_number): return [LI() for _ in range(rows_number)] def LI1(): return list(map(int1, sys.stdin.readline().split())) def LLI1(rows_number): return [LI1() for _ in range(rows_number)] def SI(): return sys.stdin.readline().rstrip() dij = [(0, 1), (-1, 0), (0, -1), (1, 0)] # dij = [(0, 1), (-1, 0), (0, -1), (1, 0), (1, 1), (1, -1), (-1, 1), (-1, -1)] # inf = -1-(-1 << 31) inf = -1-(-1 << 62) # md = 10**9+7 md = 998244353 class Sieve: def __init__(self, n): self.plist = [2] min_prime_factor = [2, 0]*(n//2+1) for x in range(3, n+1, 2): if min_prime_factor[x] == 0: min_prime_factor[x] = x self.plist.append(x) if x**2 > n: continue for y in range(x**2, n+1, 2*x): if min_prime_factor[y] == 0: min_prime_factor[y] = x self.min_prime_factor = min_prime_factor def isprime(self, x): return self.min_prime_factor[x] == x def pf(self, x): pp, ee = [], [] while x > 1: mpf = self.min_prime_factor[x] if pp and mpf == pp[-1]: ee[-1] += 1 else: pp.append(mpf) ee.append(1) x //= mpf return pp, ee # unsorted def factor(self, a): ff = [1] pp, ee = self.pf(a) for p, e in zip(pp, ee): ff, gg = [], ff w = p for _ in range(e): for f in gg: ff.append(f*w) w *= p ff += gg return ff mx = 10**6 sv = Sieve(mx) n = II() aa = LI() dp = [0]*mx ans = 0 for a in aa: if a == 1: ans += 1 continue pp, _ = sv.pf(a) odd, even = [], [] for p in pp: l, m = len(odd), len(even) for i in range(l): even.append(odd[i]*p) for i in range(m): odd.append(even[i]*p) odd.append(p) cur = 1 for b in odd: cur += dp[b] cur %= md for b in even: cur -= dp[b] cur %= md ans += cur ans %= md for b in odd: dp[b] += cur dp[b] %= md for b in even: dp[b] += cur dp[b] %= md print(ans%md)