結果

問題 No.1731 Product of Subsequence
コンテスト
ユーザー Mine
提出日時 2022-01-07 17:44:22
言語 PyPy3
(7.3.15)
結果
WA  
実行時間 -
コード長 1,215 bytes
記録
記録タグの例:
初AC ショートコード 純ショートコード 純主流ショートコード 最速実行時間
コンパイル時間 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
権限があれば一括ダウンロードができます

ソースコード

diff #
raw source code 

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)
0