結果
問題 | No.1066 #いろいろな色 / Red and Blue and more various colors (Easy) |
ユーザー |
![]() |
提出日時 | 2020-05-29 21:42:13 |
言語 | C++17 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 80 ms / 2,000 ms |
コード長 | 985 bytes |
コンパイル時間 | 1,291 ms |
コンパイル使用メモリ | 124,332 KB |
最終ジャッジ日時 | 2025-01-10 16:48:13 |
ジャッジサーバーID (参考情報) |
judge2 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include <cstdio> #include <cstring> #include <iostream> #include <string> #include <cmath> #include <bitset> #include <vector> #include <map> #include <set> #include <queue> #include <deque> #include <algorithm> #include <complex> #include <unordered_map> #include <unordered_set> #include <random> #include <cassert> #include <fstream> #include <utility> #include <functional> #include <time.h> #include <stack> #include <array> #define popcount __builtin_popcount using namespace std; typedef long long int ll; typedef pair<int, int> P; const ll MOD=998244353; int main() { int n, q; cin>>n>>q; ll a[6060]; int b[6060]; for(int i=0; i<n; i++) cin>>a[i]; for(int i=0; i<q; i++) cin>>b[i]; ll dp[2][6060]={}; dp[0][0]=1; for(int i=0; i<n; i++){ fill(dp[(i+1)&1], dp[(i+1)&1]+n+1, 0); for(int j=0; j<=i; j++){ (dp[(i+1)&1][j+1]+=dp[i&1][j])%=MOD; (dp[(i+1)&1][j]+=dp[i&1][j]*(a[i]-1))%=MOD; } } for(int i=0; i<q; i++){ cout<<dp[n&1][b[i]]<<endl; } return 0; }