結果
問題 | No.2994 べき内積 |
ユーザー |
|
提出日時 | 2024-12-21 05:45:55 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 174 ms / 2,000 ms |
コード長 | 1,465 bytes |
コンパイル時間 | 4,190 ms |
コンパイル使用メモリ | 256,312 KB |
最終ジャッジ日時 | 2025-02-26 16:01:17 |
ジャッジサーバーID (参考情報) |
judge4 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 23 |
ソースコード
#include <bits/stdc++.h> #include <atcoder/all> using namespace std; using mint = 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 mul = [&](const vector<mint> &lhs, const vector<mint> &rhs){ vector<mint> res(n); for(int i = 0; i < lhs.size(); i++){ for(int j = 0; i + j < n && j < rhs.size(); j++){ res[i + j] += lhs[i] * rhs[j]; } } return res; }; 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 = mul(tmp, B); ex /= 2; if(ex) B = mul(B, 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 = mul(ans, frobenius(K[idx++])); prd *= 1009; } mint coef = mint::raw(A[0]).pow(accumulate(K.begin() + idx, K.end(), 0)); for(int i = 0; i < n; i++){ cout << (coef * ans[i]).val() << (i + 1 == n ? '\n' : ' '); } }