結果
問題 | No.617 Nafmo、買い出しに行く |
ユーザー | hiro_metal_core |
提出日時 | 2017-12-19 21:57:27 |
言語 | Python3 (3.13.1 + numpy 2.2.1 + scipy 1.14.1) |
結果 |
AC
|
実行時間 | 527 ms / 2,000 ms |
コード長 | 1,191 bytes |
コンパイル時間 | 339 ms |
コンパイル使用メモリ | 12,544 KB |
実行使用メモリ | 10,752 KB |
最終ジャッジ日時 | 2024-12-16 02:12:01 |
合計ジャッジ時間 | 4,977 ms |
ジャッジサーバーID (参考情報) |
judge1 / judge3 |
(要ログイン)
ファイルパターン | 結果 |
---|---|
sample | AC * 2 |
other | AC * 20 |
ソースコード
N, K = list(map(int, input().split())) A = [] for n in range(N): A.append(int(input())) # 以下は全探索のアルゴリズム(TLEだった) def bin_search(item_index, rest_weight): if item_index == N: return 0 elif rest_weight < A[item_index]: return bin_search(item_index + 1, rest_weight) else: res1 = bin_search(item_index + 1, rest_weight) res2 = bin_search(item_index + 1, rest_weight - A[item_index]) + A[item_index] return max(res1, res2) print(bin_search(0, K)) # 以下は動的計画法で書いたアルゴリズム(なぜかREだった) """ import numpy as np dp = np.zeros((N + 1, K + 1)) def do_dp(): for item_index in range(N - 1, -1, -1): for rest_weight in range(K + 1): if rest_weight < A[item_index]: dp[item_index][rest_weight] = dp[item_index + 1][rest_weight] else: dp[item_index][rest_weight] = \ max(dp[item_index + 1][rest_weight], dp[item_index + 1][rest_weight - A[item_index]] + A[item_index]) return dp[0][K] ans_weight = do_dp() print(int(ans_weight)) """