結果
問題 | No.2741 Balanced Choice |
ユーザー | norioc |
提出日時 | 2024-04-25 01:19:46 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 633 ms / 2,000 ms |
コード長 | 764 bytes |
コンパイル時間 | 164 ms |
コンパイル使用メモリ | 82,176 KB |
実行使用メモリ | 77,056 KB |
最終ジャッジ日時 | 2024-04-25 01:19:54 |
合計ジャッジ時間 | 6,306 ms |
ジャッジサーバーID (参考情報) |
judge3 / judge4 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 569 ms
76,800 KB |
testcase_01 | AC | 567 ms
76,928 KB |
testcase_02 | AC | 611 ms
76,800 KB |
testcase_03 | AC | 633 ms
76,928 KB |
testcase_04 | AC | 614 ms
76,928 KB |
testcase_05 | AC | 38 ms
52,096 KB |
testcase_06 | AC | 36 ms
52,096 KB |
testcase_07 | AC | 336 ms
76,800 KB |
testcase_08 | AC | 487 ms
76,928 KB |
testcase_09 | AC | 323 ms
77,056 KB |
testcase_10 | AC | 382 ms
76,800 KB |
testcase_11 | AC | 472 ms
77,056 KB |
ソースコード
def knapsack(ss: list): dp = [-INF] * (W+1) dp[0] = 0 for w, v in ss: for i in reversed(range(W+1)): if dp[i] == -INF: continue if i+w > W: continue dp[i+w] = max(dp[i+w], dp[i] + v) return dp INF = 1 << 60 N, W, D = map(int, input().split()) ss0 = [] ss1 = [] for _ in range(N): t, w, v = map(int, input().split()) if t == 0: ss0.append((w, v)) else: ss1.append((w, v)) dp0 = knapsack(ss0) dp1 = knapsack(ss1) ans = 0 for i in range(W+1): jmax = W-i # j の最大値 lo = max(0, i-D) hi = min(i+D, jmax) for j in range(lo, hi+1): if dp0[i] == -INF: continue if dp1[j] == -INF: continue ans = max(ans, dp0[i] + dp1[j]) print(ans)