代码随想录算法训练营第四十六天

139. 单词拆分

题目链接：单词拆分
这个其实是个排列题，所以先遍历背包再遍历物品，以及因为遍历过程中，在遍历下一个物品时会出现对不上的情况，所以这种情况下，要记得加个or，就是只要有一个物品能对上就是True的。

class Solution:
    def wordBreak(self, s: str, wordDict: List[str]) -> bool:
        # dp[j] = s[:(j+1))]是否由wordDict里的单词组成
        dp = [False]*(len(s)+1)
        dp[0] = True
        
        for j in range(1,(len(s)+1)):
            for i in range(len(wordDict)):
                if (j>=len(wordDict[i])) and  s[(j-len(wordDict[i])):j] == wordDict[i]:
                    dp[j] = dp[j] or dp[j-len(wordDict[i])]
                else:
                    dp[j] = dp[j] or False
        return dp[-1]

多重背包

物品	重量	价值	数量
0	1	15	2
1	3	20	3
2	4	30	2

可以转换成0-1背包问题：
每件物品只能用一次。

物品	重量	价值
0	1	15
0	1	15
1	3	20
1	3	20
1	3	20
2	4	30
2	4	30

直接拆开

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

if __name__ == '__main__':
    test_multi_pack1()

遍历个数

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_pack2()

购物单

题目链接：购物单

主件	附件
电脑	打印机，扫描仪
书柜	图书
书桌	台灯，文具
工作椅	无

如果要买归类为附件的物品，必须先买该附件所属的主件，且每件物品只能购买一次。
每个主件可以有 0 个、 1 个或 2 个附件。附件不再有从属于自己的附件。
解题思路：

• 这个问题就是要把附件的从属信息拿出来，然后在遍历主件的时候，去看是否存在附件，如果存在，就去比较附件，如果没有就不管了。
• 如果有附件，那么需要注意，在对主件的遍历里，我们已经决定了要不要主件，是有可能不会买的，所以在往后继续遍历附件的时候要带上主件。这里如果有两个附件，那么附件的遍历就是主件+附件1，主件+附件2，主件+附件1+附件2。
• 第二个问题是两个附件之间的关系是平等的，所以if的空格不能错。

from collections import defaultdict
N,m = map(int,input().split())
primary = {}
acce = defaultdict(list)

for i in range(1,(m+1)):
    x,y,z = map(int,input().split())
    if z == 0:
        primary[i] = [x,y]
    if z != 0:
        acce[z].append([x,y])
#print(primary,acce)
dp = [0] * (N+1)
for i in primary:
    w = primary[i][0]
    v = primary[i][1]
    for j in range(N,(w-1),-1):
        if j >= w:
            dp[j] = max(dp[j],dp[j-w]+w*v) #决定主件买不买
            if i in acce:
                w1 = acce[i][0][0]
                v1 = acce[i][0][1]
                if j >= (w1+w):
                    dp[j] = max(dp[j],dp[j-w1-w]+w1*v1+w*v)
                if len(acce[i])>1:
                    w2 = acce[i][1][0]
                    v2 = acce[i][1][1]
                    if j >= (w2+w):
                        dp[j] = max(dp[j],dp[j-w2-w]+w2*v2+w*v)
                    if j >= (w1+w2+w):
                        dp[j] = max(dp[j],dp[j-w2-w1-w]+w2*v2+w1*v1+w*v)
print(dp[-1])