现有 n1 + n2 种面值的硬币,其中前 n1 种为普通币,可以取任意枚,后 n2 种为纪念币,每种最多只能取一枚,每种硬币有一个面值,问能用多少种方法拼出 m 个面值?
可能性分析
arbitrary= [ 2, 3, 5 ] , only_one= [ 1, 2, 4 ] , m = 10
- arbitrary 对 m 贡献 0 元,次数 a ,only_one 贡献 10 ,次数 b;结果为 res1 = a * b
- arbitrary 对 m 贡献 1 元,次数 c ,only_one 贡献 9 ,次数 d;结果为 res2 = c * d
- arbitrary 对 m 贡献 2 元,次数 e ,only_one 贡献 f ,次数 d;结果为 res3 = e * f
- …
- arbitrary 对 m 贡献 10 元,次数 e ,only_one 贡献 f ,次数 0;结果为 resk = x * y
- res = res1 + res2 + …+ resk
为了计算方便,我们提前计算出 arbitrary 的 dp1 和 only_one 的 dp2。在上边尝试时,只需直接查询结果即可。
注意:上边很多情况次数都是 0,比如:arbitrary 对 m 贡献 1 元,次数 c ,此时 c = 0
def coins1(arr, aim):
if not arr: return
dp = [[0] * (aim + 1) for _ in range(len(arr))]
for i in range(len(arr)):
dp[i][0] = 1
for i in range(1, aim + 1):
if arr[0] * i < aim + 1:
dp[0][arr[0] * i] = 1
for i in range(1, len(arr)):
for j in range(1, aim + 1):
dp[i][j] = dp[i - 1][j]
dp[i][j] += dp[i][j - arr[i]] if (j - arr[i]) >= 0 else 0
return dp
def coins2(arr, aim):
if not arr: return
dp = [[0] * (aim + 1) for _ in range(len(arr))]
for i in range(len(arr)):
dp[i][0] = 1
if arr[0] <= aim:
dp[0][arr[0]] = 1
for i in range(1, len(arr)):
for j in range(1, aim + 1):
dp[i][j] = dp[i - 1][j]
dp[i][j] += dp[i - 1][j - arr[i]] if j - arr[i] >= 0 else 0
return dp
def money_ways(arbitrary, only_one, money):
if money < 0: return 0
if not arbitrary and not only_one:
return 1 if money == 0 else 0
dp_1 = coins1(arbitrary, money)
dp_2 = coins2(only_one, money)
n_1 = len(arbitrary) - 1
n_2 = len(arbitrary) - 1
if not dp_1: return dp_2[n_2][money]
if not dp_2: return dp_1[n_1][money]
res = 0
for i in range(money + 1):
res += dp_1[n_1][i] * dp_1[n_2][money - i]
return res