#716149 (C++17) No.1731 Product of Subsequence

• 問題: No.1731 Product of Subsequence
• ユーザー: norikame
• 提出日時: 2021-11-13 13:41:37
• 言語: C++17 (gcc 12.3.0 + boost 1.83.0)
• 結果: AC
• 実行時間: 1,515 ms / 2,000 ms
• コード長: 1,528 bytes
• コンパイル時間: 2,602 ms
• コンパイル使用メモリ: 215,552 KB
• 実行使用メモリ: 5,248 KB
• 最終ジャッジ日時: 2024-11-08 05:11:45
• 合計ジャッジ時間: 14,573 ms
• ジャッジサーバーID （参考情報）: judge2 / judge3

入力	結果	実行時間; 実行使用メモリ
testcase_00	AC	911 ms; 5,248 KB
testcase_01	AC	2 ms; 5,248 KB
testcase_02	AC	2 ms; 5,248 KB
testcase_03	AC	2 ms; 5,248 KB
testcase_04	AC	2 ms; 5,248 KB
testcase_05	AC	3 ms; 5,248 KB
testcase_06	AC	3 ms; 5,248 KB
testcase_07	AC	2 ms; 5,248 KB
testcase_08	AC	47 ms; 5,248 KB
testcase_09	AC	4 ms; 5,248 KB
testcase_10	AC	1,492 ms; 5,248 KB
testcase_11	AC	1,283 ms; 5,248 KB
testcase_12	AC	4 ms; 5,248 KB
testcase_13	AC	18 ms; 5,248 KB
testcase_14	AC	1,181 ms; 5,248 KB
testcase_15	AC	2 ms; 5,248 KB
testcase_16	AC	3 ms; 5,248 KB
testcase_17	AC	2 ms; 5,248 KB
testcase_18	AC	471 ms; 5,248 KB
testcase_19	AC	14 ms; 5,248 KB
testcase_20	AC	386 ms; 5,248 KB
testcase_21	AC	1,515 ms; 5,248 KB
testcase_22	AC	6 ms; 5,248 KB
testcase_23	AC	3 ms; 5,248 KB
testcase_24	AC	19 ms; 5,248 KB
testcase_25	AC	8 ms; 5,248 KB
testcase_26	AC	247 ms; 5,248 KB
testcase_27	AC	324 ms; 5,248 KB
testcase_28	AC	671 ms; 5,248 KB
testcase_29	AC	250 ms; 5,248 KB
testcase_30	AC	919 ms; 5,248 KB
testcase_31	AC	6 ms; 5,248 KB
testcase_32	AC	3 ms; 5,248 KB
testcase_33	AC	6 ms; 5,248 KB
testcase_34	AC	76 ms; 5,248 KB

ソースコード

#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;
	}
	vector<pair<int, int>> kpvec;
	int ktmp = k;
	for (int i=2; i*i<=ktmp; ++i) if (ktmp%i == 0) {
		kpvec.emplace_back(i, 0);
		while (ktmp%i == 0) {
			kpvec.back().second++;
			ktmp /= i;
		}
	}
	if (ktmp > 1) kpvec.emplace_back(ktmp, 1);
	int m = kpvec.size();
	map<vector<pair<int, int>>, ll> dp;
	rep(i, n) {
		vector<pair<int, int>> aip;
		ll atmp = a[i];
		for (auto pi : kpvec) {
			aip.emplace_back(pi.first, 0);
			while (atmp%pi.first == 0) {
				aip.back().second++;
				atmp /= pi.first;
			}
			aip.back().second = min(aip.back().second, pi.second);
		}
		map<vector<pair<int, int>>, ll> ndp;
		ndp[aip] = (ndp[aip] + 1) % mod;
		for (auto pi : dp) {
			auto naip = pi.first;
			rep(j, m) naip[j].second = min(naip[j].second+aip[j].second, kpvec[j].second);
			ndp[naip] = (ndp[naip] + pi.second) % mod;
		}
		for (auto pi : ndp) dp[pi.first] = (dp[pi.first] + pi.second) % mod;
	}
	cout << dp[kpvec] << endl;
	return 0;
}