結果
問題 |
No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
|
ユーザー |
|
提出日時 | 2020-05-29 22:05:01 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 354 ms / 2,000 ms |
コード長 | 360 bytes |
コンパイル時間 | 301 ms |
コンパイル使用メモリ | 82,392 KB |
実行使用メモリ | 77,856 KB |
最終ジャッジ日時 | 2024-11-06 04:51:59 |
合計ジャッジ時間 | 4,420 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
N, Q = map(int, input().split()) A = tuple(map(int, input().split())) B = tuple(map(int, input().split())) MOD = 998244353 dp = [0] * (N + 2) dp[1] = 1 for i, a in enumerate(A): new_dp = [0] * (N + 2) for j in range(1, N + 2): new_dp[j] += dp[j] * (a - 1) + dp[j - 1] new_dp[j] %= MOD dp = new_dp for b in B: print(dp[b + 1])