結果
問題 |
No.2994 べき内積
|
ユーザー |
|
提出日時 | 2024-12-25 10:45:53 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
WA
|
実行時間 | - |
コード長 | 957 bytes |
コンパイル時間 | 2,209 ms |
コンパイル使用メモリ | 114,224 KB |
実行使用メモリ | 9,260 KB |
最終ジャッジ日時 | 2024-12-25 10:46:00 |
合計ジャッジ時間 | 6,926 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 16 WA * 7 |
ソースコード
#include<iostream> #include<vector> #include<cassert> #include<atcoder/modint> #include<atcoder/convolution> using namespace std; const int p=1009; using mint=atcoder::static_modint<p>; int M,N; int K[4<<17]; mint A[2001],comb[p][p]; vector<mint>nxt(vector<mint>a) { using m=atcoder::modint998244353; vector<m>L(N+1),R(N+1); for(int i=0;i<=N;i++)L[i]=m::raw(A[i].val()); for(int i=0;i<=N;i++)R[i]=m::raw(a[i].val()); auto M=atcoder::convolution(L,R); vector<mint>ret(N+1); for(int i=0;i<=N;i++)ret[i]=mint(M[i].val()); //for(int i=0;i<=N;i++)for(int j=0;i+j<=N;j++)ret[i+j]+=A[i]*a[j]; return ret; } int main() { ios::sync_with_stdio(false); cin.tie(nullptr); cin>>M>>N; int sum=0; for(int i=0;i<=M;i++) { cin>>K[i]; if(i)sum+=K[i]; } for(int i=0;i<=N;i++) { int a;cin>>a; A[i]=a; } sum%=p-1; vector<mint>a(N+1); a[0]=A[0].pow(sum); for(int t=1;t<=K[0];t++)a=nxt(a); for(int i=0;i<=N;i++)cout<<a[i].val()<<(i==N?"\n":" "); }