結果
問題 | No.1066 #いろいろな色 / Red and Blue and more various colors (Easy) |
ユーザー |
|
提出日時 | 2020-05-29 23:09:12 |
言語 | C++14 (gcc 13.3.0 + boost 1.87.0) |
結果 |
AC
|
実行時間 | 123 ms / 2,000 ms |
コード長 | 500 bytes |
コンパイル時間 | 1,712 ms |
コンパイル使用メモリ | 169,344 KB |
実行使用メモリ | 6,820 KB |
最終ジャッジ日時 | 2024-11-06 07:34:59 |
合計ジャッジ時間 | 4,043 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
#include<bits/stdc++.h>using namespace std;using ll = long long;const ll MOD = 998244353;ll dp[2][6001];int main(){int n,q;cin >> n >> q;vector<ll> A(n);for(int i =0;i<n;i++)cin >> A[i];dp[0][0] = 1;for(int i =0;i<n;i++){for(int j = 0;j<6000;j++){(dp[1][j] += dp[0][j]*(A[i]-1))%=MOD;(dp[1][j+1] += dp[0][j])%=MOD;}for(int j = 0;j<=6000;j++){dp[0][j] = dp[1][j];dp[1][j] = 0;}}while(q--){int b;cin >> b;cout<<dp[0][b]<<endl;}}