41 称砝码

题目描述
现有一组砝码，重量互不相等，分别为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,num):
    result=set()
    for i in range(num[0]+1):
        result.add(i*weight[0])#将第一个重量的所有砝码个数组合加入结果中，包括0
    for j in range(1,n):#砝码数量,1~n-1+1=1~n
        temp=list(result)#set转为list好遍历
        for m in range(1,num[j]+1):#取每个砝码数量
            for t in temp:#已有重量
                result.add(t+m*weight[j])#weight[j]对应的砝码质量
    return len(result)

while True:
    try:
        n=int(input())
        weight=[int(i) for i in input().split()]
        num=[int(i) for i in input().split()]
        print(fama(n,weight,num))
    except:
        break