結果
問題 |
No.1066 #いろいろな色 / Red and Blue and more various colors (Easy)
|
ユーザー |
![]() |
提出日時 | 2025-04-15 22:09:06 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 242 ms / 2,000 ms |
コード長 | 642 bytes |
コンパイル時間 | 515 ms |
コンパイル使用メモリ | 82,364 KB |
実行使用メモリ | 67,160 KB |
最終ジャッジ日時 | 2025-04-15 22:10:33 |
合計ジャッジ時間 | 3,397 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge4 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
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())) k = sum(1 for x in a if x == 1) rest = [x - 1 for x in a if x > 1] m = len(rest) # Initialize dp_rest for the rest boxes dp_rest = [0] * (m + 1) dp_rest[0] = 1 for num in rest: for j in range(m, 0, -1): dp_rest[j] = (dp_rest[j] * num + dp_rest[j - 1]) % MOD dp_rest[0] = (dp_rest[0] * num) % MOD # Process each query results = [] for query in b: t = query - k if t < 0 or t > m: results.append(0) else: results.append(dp_rest[t] % MOD) print('\n'.join(map(str, results)))