题目描述
现有一组砝码,重量互不相等,分别为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(weight[0]*i)
for i in range(1,n):
temp = list(res)
for num in range(1,nums[i]+1):
for rr in temp:
res.add(rr+weight[i]*num)
return len(res)
while True:
try:
n = int(input())
weight = [int(i) for i in input().split()]
nums = [int(i) for i in input().split()]
print(fama(n,weight,nums))
except:
break