結果
問題 | No.2025 Select -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 bisectn, 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()) - 1if q >= dp[0][-1] - dp[0][-2]:print(-1)continueres = [0] * nremaining = k + 1for 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] - offsetres[i] = remaining - jremaining = jprint(*res)