結果
| 問題 |
No.1731 Product of Subsequence
|
| コンテスト | |
| ユーザー |
|
| 提出日時 | 2022-01-07 18:00:01 |
| 言語 | PyPy3 (7.3.15) |
| 結果 |
AC
|
| 実行時間 | 1,597 ms / 2,000 ms |
| コード長 | 1,242 bytes |
| コンパイル時間 | 474 ms |
| コンパイル使用メモリ | 82,048 KB |
| 実行使用メモリ | 79,616 KB |
| 最終ジャッジ日時 | 2024-11-08 05:18:52 |
| 合計ジャッジ時間 | 20,625 ms |
|
ジャッジサーバーID (参考情報) |
judge4 / judge1 |
(要ログイン)
| ファイルパターン | 結果 |
|---|---|
| sample | AC * 4 |
| other | AC * 31 |
ソースコード
from collections import defaultdict
def prime_factorization(n):
arr = defaultdict(int)
for i in range(2, int(n**0.5)+1):
while n % i == 0:
arr[i] += 1
n //= i
if n != 1:
arr[n] += 1
return arr
def make_divisors(n):
lower_divisors , upper_divisors = [], []
i = 1
while i*i <= n:
if n % i == 0:
lower_divisors.append(i)
if i != n // i:
upper_divisors.append(n//i)
i += 1
return lower_divisors + upper_divisors[::-1]
def k_divisors(n):
ans = defaultdict(int)
for k in divmax:
while n % k == 0:
n //= k
ans[k] += 1
return ans
n, k = map(int, input().split())
a = list(map(int, input().split()))
mod = 10**9+7
divs = {}
dp = {}
for mm in make_divisors(k):
divs[mm] = prime_factorization(mm)
dp[mm] = 0
divmax = divs[k]
dp[1] += 1
for i in range(n):
divai = k_divisors(a[i])
dp_copy = dp.copy()
for fr, div in divs.items():
to = 1
for key in divmax:
to *= pow(key, min(divmax[key], div[key]+divai[key]))
dp[to] += dp_copy[fr]
dp[to] %= mod
if k == 1:
print((dp[k]-1)%mod)
else:
print(dp[k]%mod)