結果
問題 |
No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
|
ユーザー |
![]() |
提出日時 | 2023-01-30 12:43:10 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 368 ms / 2,000 ms |
コード長 | 383 bytes |
コンパイル時間 | 130 ms |
コンパイル使用メモリ | 82,020 KB |
実行使用メモリ | 78,520 KB |
最終ジャッジ日時 | 2024-06-30 01:49:07 |
合計ジャッジ時間 | 4,149 ms |
ジャッジサーバーID (参考情報) |
judge5 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
N, Q = map(int, input().split()) A = list(map(int, input().split())) B = list(map(int, input().split())) mod = 998244353 pre = [0] * (N + 5) pre[0] = 1 for i in range(N): dp = [0] * (N + 5) for j in range(N): dp[j + 1] += pre[j] dp[j] += pre[j] * (A[i] - 1) dp[j] %= mod dp[j + 1] %= mod dp, pre = pre, dp for b in B: print(pre[b])