結果

問題 No.1731 Product of Subsequence
ユーザー Mine
提出日時 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
権限があれば一括ダウンロードができます

ソースコード

diff #

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