結果
問題 |
No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
|
ユーザー |
![]() |
提出日時 | 2025-04-16 15:39:31 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 238 ms / 2,000 ms |
コード長 | 439 bytes |
コンパイル時間 | 678 ms |
コンパイル使用メモリ | 82,048 KB |
実行使用メモリ | 76,860 KB |
最終ジャッジ日時 | 2025-04-16 15:44:49 |
合計ジャッジ時間 | 3,594 ms |
ジャッジサーバーID (参考情報) |
judge4 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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())) dp = [0] * (N + 1) dp[0] = 1 for i in range(N): ai = A[i] a_minus_1 = ai - 1 new_dp = [0] * (N + 1) for j in range(i + 2): new_dp[j] = (new_dp[j] + dp[j] * a_minus_1) % MOD if j >= 1: new_dp[j] = (new_dp[j] + dp[j - 1]) % MOD dp = new_dp for b in B: print(dp[b])