結果
問題 |
No.1731 Product of Subsequence
|
ユーザー |
|
提出日時 | 2022-01-07 18:09:21 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,604 ms / 2,000 ms |
コード長 | 1,206 bytes |
コンパイル時間 | 430 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 79,360 KB |
最終ジャッジ日時 | 2024-11-08 05:18:30 |
合計ジャッジ時間 | 20,203 ms |
ジャッジサーバーID (参考情報) |
judge5 / 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): ans = [] for i in range(1, n+1): if i*i > n: break if n % i == 0: ans.append(i) if i != n // i: ans.append(n//i) return sorted(ans) 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 divmax = prime_factorization(k) divs = {} dp = {} for km in make_divisors(k): divs[km] = k_divisors(km) dp[km] = 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)