結果

問題 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;
}
הההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההההה
XXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXXX
0