题解 | #称砝码#

称砝码

https://www.nowcoder.com/practice/f9a4c19050fc477e9e27eb75f3bfd49c

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))

全部评论

相关推荐

点赞 收藏 评论
分享
牛客网
牛客企业服务