秋招笔试（其二）华为笔试箱子最大值问题

“”"
箱子最大值问题
K、N 分别代表卡车尺寸、箱子个数
每个箱子尺寸 w
每个箱子价值 v
尺寸w 和价值 v 一一对应
“”"

def dp_worth(K, v, w):
    dp = [[0 for _ in range(K+1)] for _ in range(len(w)+1)]
    for i in range(0, len(w)+1):
        for j in range(0, K+1):
            if j < w[i-1]:
                dp[i][j] = dp[i-1][j]
            else:
                A = dp[i-1][j]
                B = dp[i-1][j-w[i-1]] + v[i-1]
                dp[i][j] = max(A, B)
    return dp[len(w)][K]

if __name__ == '__main__':
    arr_w = []
    arr_v = []
    K = int(input())
    N = int(input())  # 用于输入有效数据的个数

    input_single_w = input().split(" ")            # 单次输入用 空格 隔开
    input_single_v = input().split(" ")       # 单次输入用 空格 隔开
    for i in range(N):                        # 实现一次性连续性输入n个数
        arr_w.append(int(input_single_w[i]))   # 将数字转换成整数方式存储于列表
    for j in range(N):                        # 实现一次性连续性输入n个数
        arr_v.append(int(input_single_v[j]))   # 将数字转换成整数方式存储于列表

    print(dp_worth(K, arr_v, arr_w))