华为OD机试真题 Python 实现【称砝码】

题目：

现有一组砝码，重量互不相等，分别m1,m2,m3...mn;

每种砝码对应的数量为x1,x2,x3...xn。现在要用这些砝码去称物体的重量，问能称出多少种不同的重量。

注：称重量包括0

输入：

包含多组测试数据。

对于每组测试数据：

第一行：n 砝码总数(范围[1,10])

第二行：m1 m2 m3 ... mn 每种砝码的重量(范围[1,2000])

第三行：x1 x2 x3 ... xn 每种砝码的数量(范围[1,6])

输出：

利用给定的砝码可以称出的不同的重量数

示例1：

输入

2

1 2

2 1

输出

5

result = set([])
result.add(0)

n = int(input())
weight = [int(x) for x in input().split(" ")]
nums = [int(x) for x in input().split(" ")]

for i in range(n):
    temp_result = list(result)
    for j in range(1, nums[i]+1):
        for k in temp_result:
            result.add(k + weight[i] * j)
 
print(len(result))