結果
問題 |
No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
|
ユーザー |
![]() |
提出日時 | 2025-04-15 22:12:44 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 185 ms / 2,000 ms |
コード長 | 456 bytes |
コンパイル時間 | 228 ms |
コンパイル使用メモリ | 81,912 KB |
実行使用メモリ | 68,236 KB |
最終ジャッジ日時 | 2025-04-15 22:14:06 |
合計ジャッジ時間 | 3,466 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge2 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 24 |
ソースコード
MOD = 998244353 n, q = map(int, input().split()) a = list(map(int, input().split())) b = list(map(int, input().split())) c = [x - 1 for x in a] dp = [0] * (n + 2) # dp[0..n] dp[0] = 1 current_max_k = 0 for ci in c: for k in range(current_max_k, -1, -1): dp[k + 1] = (dp[k + 1] + dp[k]) % MOD dp[k] = (dp[k] * ci) % MOD current_max_k += 1 for bi in b: if 0 <= bi <= n: print(dp[bi] % MOD) else: print(0)