砝码称重-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)