結果
| 問題 |
No.2075 GCD Subsequence
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2022-09-16 21:42:18 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
WA
|
| 実行時間 | - |
| コード長 | 1,529 bytes |
| コンパイル時間 | 465 ms |
| コンパイル使用メモリ | 82,028 KB |
| 実行使用メモリ | 106,496 KB |
| 最終ジャッジ日時 | 2024-12-21 18:50:52 |
| 合計ジャッジ時間 | 32,186 ms |
|
ジャッジサーバーID (参考情報) |
judge2 / judge3 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 2 WA * 26 |
ソースコード
from math import gcd
def isprime(n):
if n <= 1:
return False
elif n == 2:
return True
elif n % 2 == 0:
return False
A = [2, 325, 9375, 28178, 450775, 9780504, 1795265022]
s = 0
d = n - 1
while d % 2 == 0:
s += 1
d >>= 1
for a in A:
if a % n == 0:
return True
x = pow(a, d, n)
if x != 1:
for t in range(s):
if x == n - 1:
break
x = x * x % n
else:
return False
return True
def pollard(n):
if n % 2 == 0:
return 2
if isprime(n):
return n
f = lambda x:(x * x + 1) % n
step = 0
while 1:
step += 1
x = step
y = f(x)
while 1:
p = gcd(y - x + n, n)
if p == 0 or p == n:
break
if p != 1:
return p
x = f(x)
y = f(f(y))
def primefact(n):
if n == 1:
return []
p = pollard(n)
if p == n:
return [p]
left = primefact(p)
right = primefact(n // p)
left += right
return sorted(left)
MOD = 998244353
n = int(input())
A = list(map(int, input().split()))
ans = 0
dp = {}
for a in A:
se = set(primefact(a))
tot = 1
for d in se:
tot += dp.get(d, 0)
tot %= MOD
ans += tot
ans %= MOD
for d in se:
dp[d] = dp.get(d, 0) + tot
dp[d] %= MOD
print(ans % MOD)