动态规划-数塔问题

动态规划-数塔问题

如图所示，将一些数字排成数塔的形状，其中第一层有一个数字，第二层有两个数字…第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