动态规划-数塔问题
如图所示,将一些数字排成数塔的形状,其中第一层有一个数字,第二层有两个数字…第n层有n个数字,现在要从第一层走到第n层,每次只能走向下一层连接的两个数字中的一个,请问路径上所有数字和最大是多少。
这里我们以dp[i][j]来代表从第i层第j个数开始的最大路径长度,可以看出,第i层第j个数的最大路径等于第i+1层第j个数的最大路径和第i+1层第j+1个数的最大路径中较大的一个+其本身的权值即:
dp[i][j]=max(dp[i+1][j],dp[i+1][j+1])+f[i][j]
下面根据递推式写出代码:
n=int(input())#输入数塔高度
f=[]
for i in range(n):#输入数塔
tmp=[int(i) for i in input().split()]
f.append(tmp)
dp=[[0 for i in range(n)] for j in range(n)]
for j in range(n):
dp[n-1][j]=f[n-1][j]#设置边界
i=n-2
while i>=0:
for j in range(i+1):
dp[i][j]=max(dp[i+1][j],dp[i+1][j+1])+f[i][j]#根据递推式求dp数组
i-=1
print(dp[0][0])#dp[0][0]即是结果
输入数据:
5
5
8 3
12 7 16
4 10 11 6
9 5 3 9 4
输出数据:
44