結果
問題 | No.2330 Eat Slime |
ユーザー |
|
提出日時 | 2023-05-28 14:18:52 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 361 ms / 4,000 ms |
コード長 | 767 bytes |
コンパイル時間 | 2,378 ms |
コンパイル使用メモリ | 111,712 KB |
実行使用メモリ | 14,524 KB |
最終ジャッジ日時 | 2024-12-27 00:30:06 |
合計ジャッジ時間 | 11,475 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 30 |
ソースコード
#include<iostream> #include<vector> #include<algorithm> #include<atcoder/modint> #include<atcoder/convolution> using namespace std; using mint=atcoder::modint998244353; int N,M,X; int C[2<<17]; int A[2<<17],B[2<<17],Y[2<<17]; int ans[2<<17]; int main() { cin>>N>>M>>X; for(int i=0;i<N;i++)cin>>C[i]; for(int i=0;i<M;i++)cin>>A[i]>>B[i]>>Y[i]; for(int col=1;col<=5;col++) { vector<mint>idx(N,mint::raw(0)); for(int i=0;i<N;i++)if(C[i]==col)idx[i]=mint::raw(1); vector<mint>coef(N,mint::raw(0)); for(int i=0;i<M;i++)if(B[i]==col) { coef[N-A[i]]+=mint::raw(Y[i]); } vector<mint>t=atcoder::convolution(idx,coef); for(int i=N-1;i<2*N-1;i++)ans[i-(N-1)]+=t[i].val(); } int ret=0; for(int i=0;i<=N;i++)ret=max(ret,ans[i]+i*X); cout<<ret<<endl; }