华为机试——称砝码

最新推荐文章于 2024-04-20 16:42:16 发布

voidfaceless

最新推荐文章于 2024-04-20 16:42:16 发布

阅读量987

点赞数

CC 4.0 BY-SA版权

分类专栏：算法设计与分析数据结构 2019届秋招 python

本文链接：https://blog.csdn.net/voidfaceless/article/details/80966005

python 同时被 3 个专栏收录

40 篇文章

订阅专栏

算法设计与分析

34 篇文章

订阅专栏

数据结构

14 篇文章

订阅专栏

题目描述

现有一组砝码，重量互不相等，分别为m1,m2,m3…mn；
每种砝码对应的数量为x1,x2,x3...xn。现在要用这些砝码去称物体的重量，问能称出多少中不同的重量。

注：

称重重量包括0

方法原型：public static int fama(int n, int[] weight, int[] nums)

输入描述:

输入包含多组测试数据。

对于每组测试数据：

第一行：n --- 砝码数(范围[1,10])

第二行：m1 m2 m3 ... mn --- 每个砝码的重量(范围[1,2000])

第三行：x1 x2 x3 .... xn --- 每个砝码的数量(范围[1,6])

输出描述:

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

示例1

输入

2
1 2
2 1

输出

主要思路：首先将第一个砝码，从0到最大数量依次添加进set()，然后对set里面的每一个rr，分别加上剩下砝码的所有可能性，三个for循环，注意这里是set，因此不需要去重了，如果选用其他容器的话，是要考虑去重的。

def fama(n,weight,nums):
    res = set()
    for i in range(nums[0]+1):
        res.add(weight[0]*i)
    for i in range(1,n):
        temp = list(res)
        for num in range(1,nums[i]+1):
            for rr in temp:
                res.add(rr+weight[i]*num)
    return len(res)
while True:
    try:
        n = int(input())
        weight = [int(i) for i in input().split()]
        nums = [int(i) for i in input().split()]
        print(fama(n,weight,nums))
    except:
        break