結果
問題 |
No.2026 Yet Another Knapsack Problem
|
ユーザー |
![]() |
提出日時 | 2025-06-12 18:34:32 |
言語 | PyPy3 (7.3.15) |
結果 |
TLE
|
実行時間 | - |
コード長 | 1,326 bytes |
コンパイル時間 | 172 ms |
コンパイル使用メモリ | 82,780 KB |
実行使用メモリ | 133,204 KB |
最終ジャッジ日時 | 2025-06-12 18:35:24 |
合計ジャッジ時間 | 38,858 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge5 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 3 |
other | AC * 35 TLE * 1 -- * 6 |
ソースコード
import sys def main(): input = sys.stdin.read().split() ptr = 0 N = int(input[ptr]) ptr += 1 c = [] v = [] for _ in range(N): ci = int(input[ptr]) vi = int(input[ptr+1]) c.append(ci) v.append(vi) ptr += 2 INF = -10**18 dp = [[INF] * (N + 1) for _ in range(N + 1)] dp[0][0] = 0 for i in range(N): weight = i + 1 value = v[i] max_count = c[i] m = 1 remaining = max_count while remaining > 0: cnt = min(m, remaining) remaining -= cnt for k_prev in range(N, -1, -1): for w_prev in range(N, -1, -1): if dp[k_prev][w_prev] == INF: continue new_k = k_prev + cnt new_w = w_prev + cnt * weight if new_k > N or new_w > N: continue if dp[new_k][new_w] < dp[k_prev][w_prev] + cnt * value: dp[new_k][new_w] = dp[k_prev][w_prev] + cnt * value m *= 2 for k in range(1, N + 1): max_val = INF for w in range(k, N + 1): if dp[k][w] > max_val: max_val = dp[k][w] print(max_val) if __name__ == "__main__": main()