結果
| 問題 | 
                            No.1731 Product of Subsequence
                             | 
                    
| コンテスト | |
| ユーザー | 
                             | 
                    
| 提出日時 | 2022-01-07 17:44:22 | 
| 言語 | PyPy3  (7.3.15)  | 
                    
| 結果 | 
                             
                                WA
                                 
                             
                            
                         | 
                    
| 実行時間 | - | 
| コード長 | 1,215 bytes | 
| コンパイル時間 | 283 ms | 
| コンパイル使用メモリ | 82,792 KB | 
| 実行使用メモリ | 79,632 KB | 
| 最終ジャッジ日時 | 2024-11-14 03:29:18 | 
| 合計ジャッジ時間 | 19,360 ms | 
| 
                            ジャッジサーバーID (参考情報)  | 
                        judge4 / judge5 | 
(要ログイン)
| ファイルパターン | 結果 | 
|---|---|
| sample | AC * 4 | 
| other | AC * 27 WA * 4 | 
ソースコード
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.items():
        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 mm in make_divisors(k):
    divs[mm] = prime_factorization(mm)
    dp[mm] = 0
dp[1] += 1
for i in range(n):
    divk = k_divisors(a[i])
    dp_copy = dp.copy()
    for fr, div in divs.items():
        to = 1
        for key in divmax:
            to *= key**min(divmax[key], div[key]+divk[key])
        dp[to] += dp_copy[fr]
        dp[to] %= mod
print(dp[k]%mod)