《华为机试》刷题之HJ41 称砝码

一、题目

二、示例

三、代码

while True:
    try:
        n = int(input())
        m = list(map(int, input().split()))
        x = list(map(int, input().split()))
        list1 = []
        for i in range(n):
            list1.extend([m[i]]*x[i])
        result  = {0}
        for item1 in list1:
            for item2 in list(result):
                result.add(item1 + item2)
        print(len(result))
    except:
        break

四、算法说明

• 根据输入顺序，将所有的砝码用数字表示出来：将x个质量为m的砝码用数字表示，即在列表list1中存入x个值为m的数字；
• 建立空的集合result，初始化为{0}，用来存储当前砝码能产生的重量种类，并去重；
• 遍历当前砝码item1时，则将result中所有元素item2与item1进行求和(item1 + item2)之后存入集合result；
• 重复上述步骤加入所有砝码，result的元素个数即为产生的重量种类组合。

胡萝卜