华为OD机试 Python 称砝码

题目

你有一些不同的砝码，每个砝码都有自己的重量和数量。试问，用这些砝码，你可以得到多少种不同的重量组合？

输入：

第一行给出砝码的种类数 n (1 ≤ n ≤ 10)。
第二行按顺序列出每种砝码的重量，重量范围为 (1 ≤ m ≤ 2000)。
第三行按顺序列出每种砝码的数量，数量范围为 (1 ≤ x ≤ 10)。
输出：
输出一个数字，代表你可以得到的不同重量组合的数量。

示例：

输入：
2
1 2
2 1

输出：
5

说明：
用这些砝码，你可以得到重量0、1、2、3和4，共五种不同的重量。

代码

def count_weight_combinations(weight_count, weights, counts):
    # dp[i]表示重量i是否可以通过组合砝码得到
    max_weight