题解 | #称砝码#妙用集合的元素唯一性

称砝码

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

def fama(n,weight,nums):
    res = set()
    res.add(0)
    for i in range(n):
        temp = list(res)
        for j in range(nums[i]+1):
            for x in temp:
                res.add(x+j*weight[i])
    return len(res)
            

n = int(input())
weight = list(map(int, input().split()))
nums = list(map(int, input().split()))

print(fama(n,weight,nums))



全部评论

相关推荐

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