結果

問題 No.617 Nafmo、買い出しに行く
ユーザー hiro_metal_corehiro_metal_core
提出日時 2017-12-19 21:38:07
言語 Python3
(3.12.2 + numpy 1.26.4 + scipy 1.12.0)
結果
TLE  
実行時間 -
コード長 1,163 bytes
コンパイル時間 90 ms
コンパイル使用メモリ 12,544 KB
実行使用メモリ 73,140 KB
最終ジャッジ日時 2024-05-09 12:56:56
合計ジャッジ時間 6,529 ms
ジャッジサーバーID
(参考情報)
judge3 / judge4
このコードへのチャレンジ
(要ログイン)

テストケース

テストケース表示
入力 結果 実行時間
実行使用メモリ
testcase_00 TLE -
testcase_01 -- -
testcase_02 -- -
testcase_03 -- -
testcase_04 -- -
testcase_05 -- -
testcase_06 -- -
testcase_07 -- -
testcase_08 -- -
testcase_09 -- -
testcase_10 -- -
testcase_11 -- -
testcase_12 -- -
testcase_13 -- -
testcase_14 -- -
testcase_15 -- -
testcase_16 -- -
testcase_17 -- -
testcase_18 -- -
testcase_19 -- -
testcase_20 -- -
testcase_21 -- -
権限があれば一括ダウンロードができます

ソースコード

diff #

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))
"""

# 以下は動的計画法で書いたアルゴリズム
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))
0