【蓝桥杯】【动态规划】砝码称重

题目描述

你有一架天平和 N 个砝码，这 N 个砝码重量依次是 W1, W2, · · · , WN。
请你计算一共可以称出多少种不同的重量？
注意砝码可以放在天平两边。

输入

输入的第一行包含一个整数 N。
第二行包含 N 个整数：W1, W2, W3, · · · , WN。

输出

输出一个整数代表答案。

样例输入

3
1 4 6

样例输出

10

提示

【样例说明】
能称出的 10 种重量是：1、2、3、4、5、6、7、9、10、11。
1 = 1；
2 = 6 4 (天平一边放 6，另一边放 4)；
3 = 4 1；
4 = 4；
5 = 6 1；
6 = 6；
7 = 1 + 6；
9 = 4 + 6 1；
10 = 4 + 6；
11 = 1 + 4 + 6。
【评测用例规模与约定】
对于 50% 的评测用例，1 ≤ N ≤ 15。
对于所有评测用例，1 ≤ N ≤ 100，N 个砝码总重不超过 100000。

假设法码个数为n，所能称的最大重量为w，建立一个大小为（n+1）*(w+1)的二维数组。初始化的值都为零。

dp[i][j],代表当有i个法码时是否能称出重量为j的物品，如果能，则令dp[i][j] = 1。

先放大重量的砝码。

状态转移方程：

if dp[i-1][j]:

    dp[i][j] = 1       #第i个法码不放

    dp[i][j+weight[i]] =1    #第i个法码放在法码盘

    if j > m[i]:

    dp[i][j-weight[i]] =1     #第i个法码放在物品盘

最后计算sum（dp[n]）-1即可。

n = int(input())
weight = [int(i) for i in input().strip().split()]
weight.sort(reverse =True)
weight = [0] +weight
maxs = sum(weight)
dp = [[0 for j in range(maxs+1)]for i in range(n+1)]
for i in range(n+1):
    dp[i][0] = 1 
for i in range(1,n+1):
    for j in range(maxs+1):
        if dp[i-1][j] :
            dp[i][j] =1 
            dp[i][j+weight[i]]=1 
            if j>weight[i]:
                dp[i][j-weight[i]] =1 

print(sum(dp[n])-1)