結果
問題 | No.2313 Product of Subsequence (hard) |
ユーザー |
|
提出日時 | 2023-05-25 01:40:53 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 1,401 ms / 4,000 ms |
コード長 | 1,277 bytes |
コンパイル時間 | 4,352 ms |
コンパイル使用メモリ | 234,052 KB |
実行使用メモリ | 5,248 KB |
最終ジャッジ日時 | 2024-12-24 00:32:21 |
合計ジャッジ時間 | 15,844 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 27 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using ll = long long; using mint = atcoder::modint998244353; int main(){ ios::sync_with_stdio(false); cin.tie(0); int n, k; cin >> n >> k; if(k == 1){ cout << mint(2).pow(n).val() - 1 << '\n'; return 0; } vector<int> divs; for(int i = 1; i * i <= k; i++){ if(k % i == 0){ divs.emplace_back(i); divs.emplace_back(k / i); } } sort(divs.begin(), divs.end()); divs.erase(unique(divs.begin(), divs.end()), divs.end()); auto f = [&](int r, int v){ return lower_bound(divs.begin(), divs.begin() + r, v) - divs.begin(); }; int m = divs.size(); vector<mint> dp(m); vector<int> pre(m), a(n); for(int i = 0; i < n; i++){ ll v; cin >> v; a[i] = __gcd((ll)k, v); } sort(a.begin(), a.end()); dp.back() = 1; for(int i = 0; i < n; i++){ if(i >= 1 && a[i - 1] == a[i]){ for(int j = 0; j < m; j++) dp[pre[j]] += dp[j]; continue; } for(int j = 0; j < m; j++){ pre[j] = f(j + 1, divs[j] / __gcd(divs[j], a[i])); dp[pre[j]] += dp[j]; } } cout << dp[0].val() << '\n'; }