【Python】【动态规划】【神奇的口袋】

问题：

有一个神奇的口袋，总的容积是40，用这个口袋可以变出一些物品，这些物品的总体积必须是40。John现在有n个想要得到的物品，每个物品的体积分别是a1，a2……an。John可以从这些物品中选择一些，如果选出的物体的总体积是40，那么利用这个神奇的口袋，John就可以得到这些物品。现在的问题是，John有多少种不同的选择物品的方式。

输入：

输入的第一行是正整数n (1 <= n <= 20)，表示不同的物品的数目。接下来的n行，每行有一个1到40之间的正整数，分别给出a1，a2……an的值。

比如：

4（4个不同的物品）

2 （以下是每种物品的重量）

2

2

2

输出：

输出不同的选择物品的方式的数目。 比如之前的输入的输出应该为6

解题思路：

最开始直接跟着老师的过程直接写代码，感觉绕来绕去总是绕不清楚，然后自己画了一个矩阵，一下子就清楚了。

首先是通过迭代算法得到我们想要每次得到的是从前K个物品中凑出重量为W的方法的数量。每次都可以看做是选了第K个物品和没有选第K个物品的两种可能加起来的和，即ways(W,K-1)+ways(W-nums[K],K-1),前一个是没选，后一个是选了，选了就会占据包的承重，因此要减去nums[K],即物品K的重量

边界条件时当W=0时，不管K等于多少都已经凑成了刚好W的重量，因此返回1；如果W<0,则返回0，

当K=0时，如果W不等于0，就没得可以选的物品了，因此凑不成W，返回0

把迭代条件中的ways看做一个二维数组，那就可以写成ways[W][K] = ways[W][K-1] + ways[W-nums[K]][K-1].当然迭代可以从00开始算，边界条件为起始值ways[0][i]=1,(0<=i<=K),以及ways[i][0] = 0,(1<=i<=W) 最后输出ways[W][K]

以输入条件为例，可以画出二维数组应该是

代码：

def koudai(nums):
    count = [[0 for i in range(len(nums)+1)] for j in range(4+1)]
    count[0] = [1]*len(count[0])
    for i in range(1,4+1):
        for j in range(1,len(nums)+1):
            count[i][j] = count[i][j-1]
            if i-nums[j-1]>=0:
                count[i][j] += count[i-nums[j-1]][j-1]
    return count[4][len(nums)]



def main():
    m = int(input())
    nums = []
    for i in range(m):
        nums.append(int(input()))
    print(koudai(nums))


if __name__ == '__main__':
    main()

参考文献：https://blog.csdn.net/lbe19/article/details/79691391