Day 46 代码随想录单词拆分多重背包

最新推荐文章于 2024-09-08 20:02:08 发布

别倒在黎明之前QAQ

最新推荐文章于 2024-09-08 20:02:08 发布

阅读量82

点赞数

分类专栏：算法训练营文章标签： leetcode 算法动态规划

本文链接：https://blog.csdn.net/m0_61375823/article/details/129643280

版权

算法训练营专栏收录该内容

47 篇文章 0 订阅

订阅专栏

文章介绍了如何使用动态规划解决139.单词拆分问题，通过遍历单词字典并维护一个状态数组来判断给定字符串是否能由字典中的单词组成。接着，文章展示了两种方法解决多重背包问题，一种是将物品数量转换为0-1背包形式，另一种是在0-1背包的基础上增加遍历物品的数量。

摘要由CSDN通过智能技术生成

139. 单词拆分

题目链接：139. 单词拆分
在这里插入图片描述

这里强调顺序所以就先遍历背包再遍历物品。
把第一次出现的单词标记为true，如果出现了后续的单词就可以标记true。

在这里插入图片描述

class Solution(object):
    def wordBreak(self, s, wordDict):
        """
        :type s: str
        :type wordDict: List[str]
        :rtype: bool
        """
        dp = [False]*(len(s) + 1)
        dp[0] = True
        # 遍历背包
        for j in range(1, len(s) + 1):
            # 遍历单词
            for word in wordDict:
                if j >= len(word):
                    if dp[j-len(word)]==True and word == s[j - len(word):j]:
                        dp[j]=True
                    else:
                        dp[j]=dp[j]
        return dp[-1]

在这里插入图片描述

多重背包问题

题目链接：多重背包问题
在这里插入图片描述

需要将多重背包变成0-1背包，具体代码如下：

def test_multi_pack1():
    '''版本一：改变物品数量为01背包格式'''
    weight = [1, 3, 4]
    value = [15, 20, 30]
    nums = [2, 3, 2]
    bag_weight = 10
    for i in range(len(nums)):
        # 将物品展开数量为1
        while nums[i] > 1:
            weight.append(weight[i])
            value.append(value[i])
            nums[i] -= 1
    
    dp = [0]*(bag_weight + 1)
    # 遍历物品
    for i in range(len(weight)):
        # 遍历背包
        for j in range(bag_weight, weight[i] - 1, -1):
            dp[j] = max(dp[j], dp[j - weight[i]] + value[i])
    
    print(" ".join(map(str, dp)))

def test_multi_pack2():
    '''版本：改变遍历个数'''
    weight = [1, 3, 4]
    value = [15, 20, 30]
    nums = [2, 3, 2]
    bag_weight = 10

    dp = [0]*(bag_weight + 1)
    for i in range(len(weight)):
        for j in range(bag_weight, weight[i] - 1, -1):
            # 以上是01背包，加上遍历个数
            for k in range(1, nums[i] + 1):
                if j - k*weight[i] >= 0:
                    dp[j] = max(dp[j], dp[j - k*weight[i]] + k*value[i])

    print(" ".join(map(str, dp)))


if __name__ == '__main__':
    test_multi_pack1()
    test_multi_pack2()