結果
問題 |
No.2994 べき内積
|
ユーザー |
|
提出日時 | 2024-12-21 05:06:51 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 1,362 bytes |
コンパイル時間 | 4,493 ms |
コンパイル使用メモリ | 260,364 KB |
最終ジャッジ日時 | 2025-02-26 16:00:08 |
ジャッジサーバーID (参考情報) |
judge2 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 WA * 7 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using mint = atcoder::modint998244353; 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 = 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); } if(ex >= 2){ (B = convolution(B, B)).resize(n); safe(B); } ex /= 2; } if(prd == 1009){ tmp[1] = tmp[1009]; tmp.resize(2); } 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' : ' '); } }