华为机试——称砝码

题目描述

现有一组砝码，重量互不相等，分别为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

主要思路：首先将第一个砝码，从0到最大数量依次添加进set()，然后对set里面的每一个rr，分别加上剩下砝码的所有可能性，三个for循环，注意这里是set，因此不需要去重了，如果选用其他容器的话，是要考虑去重的。

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