描述
现有n种砝码,重量互不相等,分别为 m1,m2,m3…mn ;
每种砝码对应的数量为 x1,x2,x3...xn 。现在要用这些砝码去称物体的重量(放在同一侧),问能称出多少种不同的重量。
注:
称重重量包括 0
数据范围:每组输入数据满足 1 \le n \le 10 \1≤n≤10 , 1 \le m_i \le 2000 \1≤mi≤2000 , 1 \le x_i \le 10 \1≤xi≤10
输入描述:
对于每组测试数据:
第一行:n --- 砝码的种数(范围[1,10])
第二行:m1 m2 m3 ... mn --- 每种砝码的重量(范围[1,2000])
第三行:x1 x2 x3 .... xn --- 每种砝码对应的数量(范围[1,10])
输出描述:
利用给定的砝码可以称出的不同的重量数
示例1
输入:
2 1 2 2 1
复制输出:
5
复制说明:
可以表示出0,1,2,3,4五种重量。
思路:每次加一块砝码,得到所有组合,使用set去重.
对于某一个砝码重量 mim_imi,它的数量 iii每增加一个,weight
这个集合就要更新一次。
更新法则很简单:所有 weight
中的元素都与这个多出来的 mim_imi 相加,再并入原来的 weight.
这更新法则有点抽象,本质是包含了iii元素数量的组合以及小iii的所有组合
while True: try: n = int(input()) m = list(map(int, input().split())) x = list(map(int, input().split())) except: break else: counts = [] weights = {0} for i in range(n): # counts.extend(x[i]*[m[i]]) for j in range(x[i]): counts.append(m[i]) for i in counts: for j in list(weights):#此处需要使用一个常量,否则会报错 weights.add(i+j) print(len(weights))