結果
問題 |
No.2994 べき内積
|
ユーザー |
|
提出日時 | 2024-12-21 06:14:11 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 44 ms / 2,000 ms |
コード長 | 1,506 bytes |
コンパイル時間 | 4,552 ms |
コンパイル使用メモリ | 261,268 KB |
最終ジャッジ日時 | 2025-02-26 16:02:35 |
ジャッジサーバーID (参考情報) |
judge1 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using mint = atcoder::static_modint<2130706433>; using mint2 = atcoder::static_modint<1009>; int main(){ ios::sync_with_stdio(false); cin.tie(0); int m, n; cin >> m >> n; m++, n++; vector<int> K(m), A(n); for(auto &&v : K) cin >> v; for(auto &&v : A) cin >> v; vector<mint> ans(1, 1), B(n); int prd = 1, idx = 0; auto safe = [&](vector<mint> &tmp){ for(auto &&v : tmp) v = mint::raw(v.val() % 1009); }; auto frobenius = [&](int ex){ vector<mint> tmp(1, 1); for(int i = 0; i < n; i++) B[i] = mint::raw(A[i]); while(ex){ if(ex & 1){ (tmp = convolution(tmp, B)).resize(n); safe(tmp); } ex /= 2; if(ex){ (B = convolution(B, B)).resize(n); safe(B); } } if(prd == 1009){ vector<mint> res(1, tmp[0]); if(tmp.size() >= 2){ res.resize(1010); res[1009] = tmp[1]; } swap(res, tmp); } return tmp; }; while(idx < m && prd < n){ (ans = convolution(ans, frobenius(K[idx++]))).resize(n); safe(ans); prd *= 1009; } mint2 coef = mint2::raw(A[0]).pow(accumulate(K.begin() + idx, K.end(), 0)); for(int i = 0; i < n; i++){ cout << (coef * ans[i].val()).val() << (i + 1 == n ? '\n' : ' '); } }