华为机试-称砝码

文章目录

    • 描述
    • 输入描述
    • 输出描述
    • 代码解析

描述

现有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≤m
i

≤2000 , 1 \le x_i \le 10 \1≤x
i

≤10

输入描述

对于每组测试数据:
第一行:n — 砝码的种数(范围[1,10])
第二行:m1 m2 m3 … mn — 每种砝码的重量(范围[1,2000])
第三行:x1 x2 x3 … xn — 每种砝码对应的数量(范围[1,10])

输出描述

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

代码解析

n = int(input())
m = input().split()
x = input().split()
# 排列组合
# 所有砝码的重量列表,组合重量
fam = []
for i in range(n):
    fam.extend([int(m[i])] * int(x[i]))
res = {0}  # 结果集
res_1 = {}  # 中间结果集
# 使用集合取并集的方法比较方便,每加一块,相当于对已经有的重量都加一遍,然后去重
for i in fam:
    # 每增加一块砝码,所有组合都加一遍
    res_1 = {i + j for j in res}
    # 取并集去重
    res = res.union(res_1)
print(len(res))

你可能感兴趣的:(华为机试,华为机试,python)