题解 | #采药#

采药

http://www.nowcoder.com/practice/d7c03b114f0541dd8e32ce9987326c16

采药问题属于01背包问题

采用动态规划进行求解:

  • 保存结果:dp[i][j]表示前i件物品放入容量为j的背包,所能带来的最大价值
  • 递推公式:①如果第i件物品不放入背包,那么就代表我们要放的是前i - 1件物品 => dp[i][j] = dp[i - 1][j];②如果第i件物品放入背包,那么第i件物品的价值value[i]我已经获得了,除此以外,我还有j - weight[i]的空间可以放我们的前i - 1件物品;综上,可得递推公式为dp[i][j] = max(dp[i - 1][j], dp[i - 1][j - weight[i]] + value[i]
  • 空间优化:注意到上面的递推公式中,dp[i - 1]是固定不动的,因此优化成dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
  • 注意:因为j - weight[i]小于j,为了避免在访问dp[j]之前,dp[j - weight[i]]已被修改,我们需要倒序遍历
  • 最终解为:dp[m](m为背包总容量,下标范围我们设置为[0, m]
def medicineCollection():
    while True:
        try:
            list = [int(i) for i in input().split()]
            # m为背包容量(这里是采药时间),n为物品数量(这里是草药数目)
            m, n = list[0], list[1]
            # weight为物品对应的重量(这里是采摘草药所需时间),value为物品对应的价值(这里是草药的价值)
            weight, value = [], []
            for i in range(n):
                list = [int(j) for j in input().split()]
                weight.append(list[0])
                value.append(list[1])
            dp = [0 for i in range(m + 1)]  # 设置dp范围为[0, m]
            for i in range(n):
                for j in range(m, weight[i] - 1, -1): # 逆序遍历的范围为[m, weight[i]]
                    dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
            print(dp[m])
        except (EOFError, KeyboardInterrupt):
            break
            
if __name__ == '__main__':
    medicineCollection()
全部评论

相关推荐

海康威视 软开岗 15k15
点赞 评论 收藏
分享
Noob1024:一笔传三代,人走笔还在
点赞 评论 收藏
分享
点赞 收藏 评论
分享
牛客网
牛客企业服务