问题:
给一个数组,数组中每一个位置的值都代表一枚钱币,求钱币任意组合,需要满足最后的金额是rest时,有多少种方法
array = [2,1,4,5]
rest = 10
暴力尝试版本
def ways1(array: List, rest: int):
return process(array, 0, rest)
def process(array, index, rest):
if index == len(array):
return 1 if rest == 0 else 0
ways = 0
zhang = 0
while rest - array[index] * zhang >= 0:
ways += process(array, index + 1, rest - array[index] * zhang)
zhang += 1
return ways
动态规划版本
- 已知当index == len(array)时,rest==0 时是1,其他全是0
index\rest | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 |
---|---|---|---|---|---|---|---|---|---|---|---|
0 | |||||||||||
1 | |||||||||||
2 | |||||||||||
3 | |||||||||||
4 | 1 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 | 0 |
- 根据递归调用可知,当前行总是需要后面一行的数据,来计算(第三行需要第四行的数据)
- 假设当前要求 (3,7)位置的值,总是需要 (4,7)位置,也就是正下方的值
- 因为还有枚举张数的行为,所以还需要 rest - array[index] * zhang 的值 也就是还需要(4,2)的值
index\rest | 0 | 1 | 2 | 3 | 4 | 5 | 6 | 7 | 8 | 9 | 10 |
---|---|---|---|---|---|---|---|---|---|---|---|
0 | |||||||||||
1 | |||||||||||
2 | |||||||||||
3 | ? | ||||||||||
4 | 1 | 0 | 0 ✔ | 0 | 0 | 0 | 0 | 0 ✔ | 0 | 0 | 0 |
def dp1(array, rest):
N = len(array)
dp = [[0 for j in range(rest + 1)] for i in range(N + 1)]
dp[N][0] = 1
index = N - 1
while index >= 0:
for i in range(rest + 1):
zhang = 0
while i - array[index] * zhang >= 0:
dp[index][i] += dp[index + 1][i - array[index] * zhang]
zhang += 1
index -= 1
return dp[0][rest]
def ways2(array: List, rest: int):
return dp1(array, rest)
利用斜率进行优化
array = [2,1,4,5]
rest = 10
-
假设需要求(0,8)这个位置的值,按照动态规划是需要得到 (1,8)、(1,6)、(1,4)、(1,2)、(1,0)的值,但是经过观察,其实(1,6)、(1,4)、(1,2)、(1,0)的值的和在 (0,6)这个位置已经求过了,所以只需要 (1,8) + (0,6)位置的值,就可以得到 (0,8)位置的值
index\rest 0 1 2 3 4 5 6 7 8 9 10 0 ? 1 ✔ ✔ ✔ ✔ ✔ 2 3 4 1 0 0 0 0 0 0 0 0 0 0
def dp2(array, rest):
N = len(array)
dp = [[0 for j in range(rest + 1)] for i in range(N + 1)]
dp[N][0] = 1
index = N - 1
while index >= 0:
for i in range(rest + 1):
dp[index][i] = dp[index + 1][i]
if i - array[index] >= 0:
dp[index][i] += dp[index][i - array[index]]
index -= 1
return dp[0][rest]
def ways3(array: List, rest: int):
return dp2(array, rest)