蓝桥杯砝码称重-dp-python题解

砝码称重-dp

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

input：
3
1 4 6
output：
10

number = int(input())  # 砝码数量
arr = list(map(int, input().split()))
summ = sum(arr)
# 动态规划数组  dp[i][j]代表加入第i个砝码时，能不能称出重量j
dp = [[False for l in range(summ + 1)] for h in range(number + 1)]
 
dp[1][arr[0]] = True  # 第一个砝码的重量肯定可以被秤出来
for i in range(2, number + 1):  # 从第二个砝码开始
    for j in range(1, summ + 1):  # 复制上一组的状态
        dp[i][j] = dp[i - 1][j]
    weight = arr[i - 1]  # 当前砝码的重量
    dp[i][weight] = True  # 当前重量可以被称量
    for j in range(1, summ + 1):  # 在第一砝码的状态下更行第二个砝码的状态
        if dp[i - 1][j]:  # 如果第一个砝码能称出重量j  在此基础上加减
            dp[i][j + weight] = True
            dp[i][abs(j - weight)] = True
 
count = 0
for j in range(1, summ + 1):
    if dp[number][j]:
        count += 1
print(count)