結果
問題 | No.1731 Product of Subsequence |
ユーザー |
|
提出日時 | 2021-11-05 22:20:36 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 1,112 ms / 2,000 ms |
コード長 | 809 bytes |
コンパイル時間 | 442 ms |
コンパイル使用メモリ | 82,432 KB |
実行使用メモリ | 78,208 KB |
最終ジャッジ日時 | 2024-11-08 04:44:43 |
合計ジャッジ時間 | 12,779 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 4 |
other | AC * 31 |
ソースコード
import sys input = sys.stdin.readline from collections import * from math import gcd N, K = map(int, input().split()) A = list(map(int, input().split())) MOD = 10**9+7 if K==1: exit(print((pow(2, N, MOD)-1)%MOD)) divs = [] for d in range(1, int(K**0.5)+1): if K%d==0: divs.append(d) if d!=K//d: divs.append(K//d) divs.sort() idx = defaultdict(int) for i in range(len(divs)): idx[divs[i]] = i M = len(idx.keys()) dp = [0]*M dp[0] = 1 for i in range(N): ndp = [0]*M for j in range(M): ndp[j] += dp[j] ndp[j] %= MOD if divs[j]*A[i]%K==0: nj = M-1 else: nj = idx[gcd(K, divs[j]*A[i])] ndp[nj] += dp[j] ndp[nj] %= MOD dp = ndp print(dp[-1])