結果

問題 No.1731 Product of Subsequence
ユーザー norikame
提出日時 2021-11-12 17:57:26
言語 C++17
(gcc 13.3.0 + boost 1.87.0)
結果
TLE  
実行時間 -
コード長 1,019 bytes
コンパイル時間 2,411 ms
コンパイル使用メモリ 208,816 KB
最終ジャッジ日時 2025-01-25 15:25:45
ジャッジサーバーID
（参考情報） 		judge8 / judge5
このコードへのチャレンジ
（要ログイン）
ファイルパターン 結果
sample AC * 4
other AC * 11 TLE * 20
権限があれば一括ダウンロードができます

ソースコード

diff # 

#include <bits/stdc++.h>
using namespace std;

using ll = long long;

#define rep(i, n) for (int i=0; i<(int)(n); ++(i))
#define rep3(i, m, n) for (int i=(m); (i)<(int)(n); ++(i))
#define repr(i, n) for (int i=(int)(n)-1; (i)>=0; --(i))
#define rep3r(i, m, n) for (int i=(int)(n)-1; (i)>=(int)(m); --(i))
#define all(x) (x).begin(), (x).end()

// 

const ll mod = (ll)(1e9) + 7;

int main() {
	int n, k;
	cin >> n >> k;
	vector<ll> a(n);
	rep(i, n) cin >> a[i];
	if (k == 1) {
		ll res = 1;
		rep(i, n) res = res * 2 % mod;
		res = (res - 1 + mod) % mod;
		cout << res << endl;
		return 0;
	}
	unordered_set<int> kprimes;
	for (int i=1; i*i<=k; ++i) if (k%i == 0) {
		kprimes.insert(i);
		kprimes.insert(k/i);
	}
	unordered_map<int, ll> dp;
	for (int pval : kprimes) dp[pval] = 0;
	rep(i, n) {
		auto ndp = dp;
		for (auto pi : ndp) dp[(int)((ll)pi.first*(a[i]%k)%k)] = (dp[(int)((ll)pi.first*(a[i]%k)%k)] + pi.second) % mod;
		dp[(int)(a[i]%k)] = (dp[(int)(a[i]%k)] + 1) % mod;
	}
	cout << dp[0] << endl;
	return 0;
}
0