結果
問題 |
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)