砝码称重 python题解

1.题目描述

题目链接：“蓝桥杯”练习系统

2.解题思路

动态规划

dp[i]：选择前i个砝码

dp[i][j]：前i个砝码可以称出重量j

dp[i][j] = 0 表示前i个砝码不可以称出重量j

dp[i][j] = 1 表示前i个砝码可以称出重量j

3. 代码实现

n = int(input())
w = list(map(int,input().split()))
w = [0] +w
dp = [[0]*(sum(w)+1) for _ in range(n+1)]
dp[0][0] = 1

for i in range(1,n+1):
    for j in range(sum(w)+1):
        if dp[i-1][j] == 1:
            dp[i][j] = 1
            dp[i][j+w[i]] = 1
            dp[i][abs(j-w[i])] = 1
print(sum(dp[n])-1)

 4. 代码解读

dp[0][0]：0个砝码可以称出重量0

sum(w) 表示最多可以称出的重量

for i in range(1,n+1) 表示循环选择前i个砝码

for j in range(sum(w)+1) 表示循环验证前i个砝码能否称出重量j

  if dp[i-1][j] == 1: 如果前i-1个砝码可以称出重量j，

① 那么前i个砝码一定也可以称出重量j，即不将第i个砝码放在天平上，所以dp[i][j] = 1

② 第i个砝码放在天平上，分两种情况：

        · 和前i-1个砝码放在同一侧，可称出重量j+w[i]，则dp[i][j+w[i]] = 1

        · 和前i-1个砝码放在不同侧，可称出重量abs(j-w[i])，则dp[i][abs(j-w[i])] = 1

 sum(dp[n])-1：sum(dp[n])表示将n个砝码可称出重量的个数加起来，减一表示除去dp[n][0]，即除去n个砝码都不放在天平上的情况。