結果
問題 | No.2741 Balanced Choice |
ユーザー | norioc |
提出日時 | 2024-04-25 00:42:56 |
言語 | PyPy3 (7.3.15) |
結果 |
AC
|
実行時間 | 727 ms / 2,000 ms |
コード長 | 754 bytes |
コンパイル時間 | 352 ms |
コンパイル使用メモリ | 82,560 KB |
実行使用メモリ | 77,312 KB |
最終ジャッジ日時 | 2024-04-25 00:43:04 |
合計ジャッジ時間 | 7,362 ms |
ジャッジサーバーID (参考情報) |
judge2 / judge1 |
(要ログイン)
テストケース
テストケース表示入力 | 結果 | 実行時間 実行使用メモリ |
---|---|---|
testcase_00 | AC | 626 ms
76,928 KB |
testcase_01 | AC | 616 ms
76,928 KB |
testcase_02 | AC | 692 ms
76,800 KB |
testcase_03 | AC | 727 ms
76,928 KB |
testcase_04 | AC | 674 ms
76,800 KB |
testcase_05 | AC | 42 ms
51,968 KB |
testcase_06 | AC | 42 ms
51,840 KB |
testcase_07 | AC | 385 ms
76,800 KB |
testcase_08 | AC | 541 ms
77,312 KB |
testcase_09 | AC | 341 ms
76,928 KB |
testcase_10 | AC | 420 ms
77,056 KB |
testcase_11 | AC | 541 ms
76,672 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): lo = max(0, i-D) hi = min(i+D, W) for j in range(lo, hi+1): if i+j > W: break if dp0[i] == -INF: continue if dp1[j] == -INF: continue ans = max(ans, dp0[i] + dp1[j]) print(ans)