题解 | #称砝码#

称砝码

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

# 定义函数计算可称出的不同重量总数
def countDifferentWeights(n, m, x):
    # 计算所有砝码可能组成的最大重量
    max_weight = sum(m[i] * x[i] for i in range(n))
    # 初始化动态规划数组,记录每个重量是否可达
    dp = [False] * (max_weight + 1)
    # 重量0总是可达的
    dp[0] = True

    # 遍历每种砝码
    for i in range(n):
        # 根据每种砝码的数量,重复更新dp数组
        for _ in range(x[i]):
            # 从最大可能重量向下遍历至该砝码重量
            for weight in range(max_weight, m[i] - 1, -1):
                # 如果当前重量减去该砝码重量是可达的,则更新当前重量为可达
                if dp[weight - m[i]]:
                    dp[weight] = True

    # 统计所有可达的重量数
    return sum(dp)


# 输入处理循环,连续读取输入直到EOF(文件结束标志)
while True:
    try:
        # 读取砝码种数
        n = int(input())
        # 读取每种砝码的重量
        m = list(map(int, input().split()))
        # 读取每种砝码的数量
        x = list(map(int, input().split()))
        # 调用函数并打印可称出的不同重量总数
        print(countDifferentWeights(n, m, x))
    except EOFError:
        # 当遇到文件结束标志时,退出循环
        break

全部评论

相关推荐

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