結果
問題 | No.2025 Select $k$-th Submultiset |
ユーザー |
![]() |
提出日時 | 2022-07-09 10:09:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 572 ms / 2,000 ms |
コード長 | 774 bytes |
コンパイル時間 | 260 ms |
コンパイル使用メモリ | 82,248 KB |
実行使用メモリ | 84,608 KB |
最終ジャッジ日時 | 2024-07-05 23:49:56 |
合計ジャッジ時間 | 15,910 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 42 |
ソースコード
from bisect import bisect n, k = map(int, input().split()) c = list(map(int, input().split())) # dp[i][j] := c[i:]を使って長さj未満の部分列を作る場合の通り数 dp = [[0] + [1] * (k + 1) for _ in range(n + 1)] for i in reversed(range(n)): for j in range(1, k + 2): dp[i][j] = dp[i + 1][j] - dp[i + 1][max(j - c[i] - 1, 0)] + dp[i][j - 1] m = int(input()) for _ in range(m): q = int(input()) - 1 if q >= dp[0][-1] - dp[0][-2]: print(-1) continue res = [0] * n remaining = k + 1 for i in range(n): offset = dp[i + 1][max(remaining - c[i] - 1, 0)] j = bisect(dp[i + 1], q + offset) q -= dp[i + 1][j - 1] - offset res[i] = remaining - j remaining = j print(*res)