美团2019秋招笔试:考试策略

本文深入解析了背包问题的两种代码实现方法,一种是传统的二维动态规划,另一种则通过优化空间复杂度来提高效率。通过对比两种实现方式,读者可以更好地理解背包问题的解决思路和优化技巧。

摘要生成于 C知道 ,由 DeepSeek-R1 满血版支持, 前往体验 >

题目描述:

题目地址:https://www.nowcoder.com/questionTerminal/a1792d443f914f2b928d2a157cd7900d

我的代码实现:

n = int(input())
k = 120
w, v = [], []
for _ in range(n):
    w1, v1, w2, v2 = [int(i) for i in input().split()]
    w.append([w1, w2])
    v.append([v1, v2])

dp = [[0 for i in range(k + 1)] for j in range(n + 1)]
for i in range(1, n + 1):
    for j in range(1, k + 1):
        dp[i][j] = dp[i-1][j]
        if j >= w[i-1][0] or j >= w[i-1][1]:
            if j >= w[i-1][0] and j >= w[i-1][1]:
                dp[i][j] = max(dp[i][j], dp[i-1][j-w[i-1][0]] + v[i-1][0], dp[i-1][j-w[i-1][1]] + v[i-1][1])
            elif j >= w[i-1][0]:
                dp[i][j] = max(dp[i][j], dp[i-1][j-w[i-1][0]] + v[i-1][0])
            else:
                dp[i][j] = max(dp[i][j], dp[i-1][j-w[i-1][1]] + v[i-1][1])
print(dp[-1][-1])

别人的代码实现(减少了空间复杂度):

n = int(input())
weight = []
value = []
for i in range(n):
    w1, v1, w2, v2 = map(int, input().split())
    weight.append([w1, w2])
    value.append([v1, v2])
 
dp = [0]*121
for i in range(n):
    for j in range(120,0, -1):
        if j >= weight[i][0] or j >= weight[i][1]:
            if j >= weight[i][0] and j >= weight[i][1]:
                dp[j] = max(dp[j-weight[i][0]]+value[i][0], dp[j-weight[i][1]]+value[i][1], dp[j])
            elif j >= weight[i][0]:
                dp[j] = max(dp[j-weight[i][0]]+value[i][0], dp[j])
            else:
                dp[j] = max(dp[j-weight[i][1]]+value[i][1], dp[j])
print(dp[120])

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值