动态规划——经典数字三角形
题目:在数字三角形中寻找一条从顶部到底部的路径,使得路径上所经过的数字之和最大。路径上的每一步都只能往左下或由下走。求出最大和,不必给出具体路径。
三角形的行数大于1小于等于100,数字为0~99.
输入格式:
5
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
(第一行是三角形行数,下面是数字三角形)
解题思路1:
自上而下,用 dp[i][j] 表示走到第 i 行第 j 列的时候最大路径的值。
7
3 8
8 1 0
2 7 4 4
4 5 2 6 5
dp 赋值:
7 0 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
第一步:
7 0 0 0 0 + 3 8
10 15 0 0 0
0 0 0 0 0
0 0 0 0 0
0 0 0 0 0
第二步:
7 0 0 0 0
10 15 0 0 0 + 8 1 0
18 16 15 0 0
0 0 0 0 0
0 0 0 0 0
第三步:
7 0 0 0 0
10 15 0 0 0
18 16 15 0 0 + 2 7 4 4
20 25 20 19 0
0 0 0 0 0
第四步:
7 0 0 0 0
10 15 0 0 0
18 16 15 0 0
20 25 20 19 0 + 4 5 2 6 5
24 30 27 26 24
n = int(input())
m = []
for i in range(n):
m.append([int(i) for i in input().split()])
dp = [[0 for j in range(n)] for i in range(n)]
dp[0][0] = m[0][0]
for i in range(n):
for j in range(i+1):
dp[i][j] = max(dp[i-1][j],dp[i-1][j-1])+m[i][j]
print(max(dp[-1]))
这里输出 dp 即为最后结果,最后一行的最大值是输出值。
解题思路2:
自下而上,直接在原来的数组上做动态规划,最后一行不动,上面一行取当前位置的最大值。
第一步:
7
3 8
8 1 0
2 7 4 4 ——> 7 12 10 10
4 5 2 6 5
(7=max(2+4,2+5), 12=max(7+5,7+2), 10=max(4+2,4+6), 10=max(4+6,4+5))
第二步:
7
3 8
8 1 0 ——> 20 13 10
7 12 10 10
4 5 2 6 5
(20=max(8+7,8+12), 13=max(1+12,1+10), 10=max(0+10, 0+10))
第三步:
7
3 8 ——> 23 21
20 13 10
7 12 10 10
4 5 2 6 5
第四步:
7 ——> 30
23 21
20 13 10
7 12 10 10
4 5 2 6 5
n = int(input())
m = []
for i in range(n):
m.append([int(i) for i in input().split()])
for r in range(n-2, -1, -1):
for i in range(r+1):
m[r][i] = max(m[r+1][i],m[r+1][i+1])+m[r][i]
print(m[0][0])
输出结果:30
这里直接输出 m,就是计算结果,第一行第一列即最终结果。