結果
| 問題 |
No.2313 Product of Subsequence (hard)
|
| ユーザー |
sotanishy
|
| 提出日時 | 2023-05-24 21:39:16 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
RE
|
| 実行時間 | - |
| コード長 | 833 bytes |
| コンパイル時間 | 716 ms |
| コンパイル使用メモリ | 82,432 KB |
| 実行使用メモリ | 138,184 KB |
| 最終ジャッジ日時 | 2024-12-23 21:52:31 |
| 合計ジャッジ時間 | 31,333 ms |
|
ジャッジサーバーID (参考情報) |
judge5 / judge4 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 3 |
| other | AC * 24 RE * 3 |
ソースコード
from math import gcd
import sys
input = sys.stdin.readline
def divisor(n):
divisors = []
i = 1
while i * i < n:
if n % i == 0:
divisors.append(i)
divisors.append(n // i)
i += 1
if i * i == n:
divisors.append(i)
divisors.sort()
return divisors
N, K = map(int, input().split())
A = list(map(int, input().split()))
if K == 1:
print((pow(2, N, mod) - 1) % mod)
exit()
div = divisor(K)
idx = {d: i for i, d in enumerate(div)}
D = len(div)
prod_idx = [[0]*D for _ in range(D)]
for i in range(D):
for j in range(D):
prod_idx[i][j] = idx[gcd(div[i]*div[j], K)]
mod = 998244353
dp = [0]*D
dp[0] = 1
for x in A:
i = idx[gcd(x, K)]
for j in range(D)[::-1]:
k = prod_idx[i][j]
dp[k] = (dp[k]+dp[j]) % mod
print(dp[-1])
sotanishy