题目描述
现有一组砝码,重量互不相等,分别为m1,m2,m3…mn;
每种砝码对应的数量为x1,x2,x3...xn。现在要用这些砝码去称物体的重量(放在同一侧),问能称出多少种不同的重量。
注:
称重重量包括0
方法原型:public static int fama(int n, int[] weight, int[] nums)
输入描述:
输入包含多组测试数据。 对于每组测试数据: 第一行:n --- 砝码数(范围[1,10]) 第二行:m1 m2 m3 ... mn --- 每个砝码的重量(范围[1,2000]) 第三行:x1 x2 x3 .... xn --- 每个砝码的数量(范围[1,6])
输出描述:
利用给定的砝码可以称出的不同的重量数
示例1
输入
复制
2 1 2 2 1
输出
复制
5
def fama(n,weight,nums):
res = set()
for i in range(nums[0]+1): #先存第一个砝码各种数量的情况
res.add(i*weight[0]) #包括0重量情况
for i in range(1,n): #砝码不同重量遍历
tmp = list(res) #能称出的体重不重复
for j in range(1,nums[i]+1): #砝码对应不同个数的遍历
for wt in tmp: #遍历已有的重量,加上新的重量
res.add(wt+j*weight[i])
return len(res)
n = int(input())
weight = [int(i) for i in input().split()] #重量
nums = [int(i) for i in input().split()] #砝码个数
print(fama(n,weight,nums)) #求能称出多少不同重量